【PyTorch】ValueError: can‘t optimize a non-leaf Tensor

端木志诚
2023-12-01

报错原因.

  • 定义张量时类型为 C P U \rm CPU CPU,将其 r e q u i r e s _ g r a d \rm requires\_grad requires_grad 设置为 T r u e \rm True True 后执行 . c u d a ( ) \rm .cuda() .cuda() 操作,报如题所示错误。
opt.requires_grad = True
opt = opt.cuda()

解决方案.

  • 需要先执行 . c u d a ( ) \rm .cuda() .cuda(),再设置自动微分属性。
opt = opt.cuda()
opt.requires_grad = True
 类似资料: