摘要: 训练神经网络时,最常用的算法就是反向传播。在该算法中,参数(模型权重)会根据损失函数关于对应参数的梯度进行调整。 为了计算这些梯度,PyTorch内置了名为 torch.autograd 的微分引擎。它支持任意计算图的自动梯度计算。 一个最简单的单层神经网络,输入 x,参数 w 和 b,某个损失函数 阅读全文
posted @ 2022-02-08 09:22 Deep_RS 阅读(531) 评论(0) 推荐(0) 编辑