search

ShusenTang / Dive-into-DL-PyTorch

本项目将《动手学深度学习》(Dive into Deep Learning)原书中的MXNet实现改为PyTorch实现。
http://tangshusen.me/Dive-into-DL-PyTorch
Apache License 2.0
18.32k stars 5.4k forks source link

3.3 节最后的损失函数需要sum后才可以调用backward函数 #83

Open kevindragon opened 4 years ago

kevindragon commented 4 years ago

3.3.7 训练模型：损失函数需要sum

ShusenTang commented 4 years ago

MSELoss默认会求平均，所以得到的loss已经是个标量。