PyTorch学习系列(五)——自定义loss

来源:互联网 发布:网络b类违规是什么意思 编辑:程序博客网 时间:2024/05/17 22:00

在PyTorch中可以自定义loss,并且只要输入到输出的过程中一直使用的都是Variable(没有用tensor或者numpy),那么loss可以自动求梯度。

那么如何确定定义的loss对各参数的梯度是否正确呢?

原创粉丝点击