本文主要是介绍【已解决】backward() got an unexpected keyword argument 'retain_variables',希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
出现这个报错的原因是,在新版本的Python下,原来的变量retain_variables
被替换成了retain_graph
。因此,在调用位置将其替换就可以了。
注:retain_graph的作用是为了避免内存在反向传播之后被释放,从而导致的无法第二次反向传播。因此需要将backward(retain_graph=True)。
更多内容,欢迎加入星球讨论。
这篇关于【已解决】backward() got an unexpected keyword argument 'retain_variables'的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!