retain_variables报错，替换成retain_graph
L_img.backward(retain_graph=True) # in case of current version of pytorch
# L_img.backward(retain_variables=True)
PyTorch0.4 中 backward()
猜你喜欢
转载自blog.csdn.net/qq_35608277/article/details/85333556
今日推荐
周排行