Torch.no_grad()影响MSE损失

2024年5月8日 153次阅读来源: pytorch

今天在训练网络的时候，发现mseloss在train和test时相差了好几个量级，后来发现是因为mseloss(input,target)，并不是对称的，input的应该是网络的输出，而targte是目标，如果颠倒过来，计算就是错的。

    原文作者：pytorch
    原文地址: https://www.cnblogs.com/elliottzheng/p/10330019.html
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。