深度学习
文章平均质量分 93
CHRN晨
如今你的气质里,藏着你走过的路,读过的书和你爱过的人。
展开
-
史上最全pytorch学习笔记2完结【文末附有源代码和参考书籍】
Module 类是 nn 模块⾥提供的⼀个模型构造类,是所有神经⽹络模块的基类,我们可以继承它来定义我们想要的模型。下⾯继承 Module 类构造本节开头提到的多层感知机。这⾥定义的 MLP 类加载了Module 类的函数和 forward 函数。它们分别⽤于创建模型参数和定义前向计算。前向计算也即正向传播。以上的 MLP 类中⽆须定义反向传播函数。系统将通过⾃动求梯度⽽⾃动⽣成反向传播所需的backward 函数。我们可以实例化 MLP 类得到模型变量 net。原创 2024-03-26 18:31:00 · 709 阅读 · 0 评论 -
史上最全pytorch学习笔记1【附有源代码和参考书籍】
此外,还可以⽤ with torch.no_grad() 将不想被追踪的操作代码块包裹起来,这种⽅法在评估模型的时候很常⽤,因为在评估模型时,我们并不需要计算可训练参数( requires_grad=True )的梯度。每个 Tensor 都有⼀个 .grad_fn 属性,该属性即创建该 Tensor 的Function , 就是说该 Tensor 是不是通过某些运算得到的,若是,则 grad_fn 返回⼀个与这些运算相关的对象,否则是None。在另⼀类情景中,模型输出可以是⼀个像图像类别这样的离散值。原创 2024-03-26 18:29:49 · 1038 阅读 · 0 评论