请问老师loss如何关联到网络的参数的

来源:4-4 利用神经网络解决分类和回归问题(2)

HelloMrDeng

2021-05-09

   # loss
   loss_func = torch.nn.MSELoss()
   loss = loss_func(pred, y_data)
   # 反向传播更新参数梯度
   loss.backward()

请问老师loss如何跟网络的梯度值进行关联的,感觉没有把网络传进去呀,怎么loss.backward()就更新了网络的权重和偏置了

写回答

1回答

会写代码的好厨师

2021-05-11

反向传播的过程实际上是pytorch内部直接实现的。具体的计算原理是使用链式法则,梯度求导去。根据求导出来的结果和梯度值,计算网络对应的参数值。

0
1
HelloMrDeng
谢谢老师!
2021-05-17
共1条回复

PyTorch入门到进阶 实战计算机视觉与自然语言处理项目

理论基础+技术讲解+实战开发,快速掌握PyTorch框架

1190 学习 · 293 问题

查看课程