【已解決】backward() got an unexpected keyword argument 'retain_variables'

出現這個報錯的原因是,在新版本的Python下,原來的變量retain_variables被替換成了retain_graph。因此,在調用位置將其替換就可以了。

注:retain_graph的作用是爲了避免內存在反向傳播之後被釋放,從而導致的無法第二次反向傳播。因此需要將backward(retain_graph=True)。

更多內容,歡迎加入星球討論。
在這裏插入圖片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章