本文主要是介绍RuntimeError:the derivative for ‘weight’ is not implemented,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
RuntimeError:the derivative for ‘weight’ is not implemented
原因:binary_cross_entropy_with_logits doesn’t support back-propagating through the weight attribute
解决方案:If you don’t need the derivative weight then you can use weight.detach() instead of weight. If you need the derivative, then you’ll having to implement binary_cross_entropy_with_logits yourself.
这篇关于RuntimeError:the derivative for ‘weight’ is not implemented的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!