原来的代码是.to(device).requires_grad_(True)
改成.requires_grad_(True).to(device)
即可正常运行
ValueError: can‘t optimize a non-leaf Tensor
最新推荐文章于 2024-07-12 19:06:42 发布
原来的代码是.to(device).requires_grad_(True)
改成.requires_grad_(True).to(device)
即可正常运行