【pytorch: can't optimize a non-leaf Tensor】

最新推荐文章于 2024-07-09 10:16:35 发布

npkhgl

最新推荐文章于 2024-07-09 10:16:35 发布

阅读量1.6w

点赞数 23

分类专栏：问题记录文章标签： pytorch non-leaf

本文链接：https://blog.csdn.net/nkhgl/article/details/100047276

版权

问题记录专栏收录该内容

13 篇文章 0 订阅

订阅专栏

需要先把Tensor放入到GPU中，然后再设置Tensor.requires_grad=True。

from torch import optim

weights = torch.rand(2,1,128,416)
weights.requires_grad = True
weights = weights.cuda()
optimizer = optim.Adam([weights], lr=0.01)
.
.
.

这样写会报“can't optimize a non-leaf Tensor”的错。

from torch import optim

weights = torch.rand(2,1,128,416)
weights = weights.cuda()
weights.requires_grad = True
optimizer = optim.Adam([weights], lr=0.01)
.
.
.

这样写就好了。