出现这个报错的原因是,在新版本的Python下,原来的变量retain_variables
被替换成了retain_graph
。因此,在调用位置将其替换就可以了。
注:retain_graph的作用是为了避免内存在反向传播之后被释放,从而导致的无法第二次反向传播。因此需要将backward(retain_graph=True)。
更多内容,欢迎加入星球讨论。
出现这个报错的原因是,在新版本的Python下,原来的变量retain_variables
被替换成了retain_graph
。因此,在调用位置将其替换就可以了。
注:retain_graph的作用是为了避免内存在反向传播之后被释放,从而导致的无法第二次反向传播。因此需要将backward(retain_graph=True)。
更多内容,欢迎加入星球讨论。