训练的时候 代码总搞不对，想请教一下怎么回事？

王义峰华  

2017-11-28

训练的时候 Pytorch代码总搞不对，想请教一下怎么回事？

 

提示错误：

AssertionError: nn criterions don't compute the gradient w.r.t. targets - please mark these variables as volatile or not requiring gradient

 

代码如下：

 

interpolation()接收两张图片，进行卷积...

各位大神帮我看下 谢谢！

 

 收藏  分享
机器学习 深度学习 PyTorch 神经网络
关注问题 3人已关注
 写回答 1人已回答
1个回答

暂时没有回答，开始写第一个回答