python中backward函数_Pythorch:运行时错误:函数MulBackward0在索引0处返回了一个无效的渐变,预期类型为torch.cuda.FloatTensor但是得到了火炬。浮雕...
我不明白这个错误在告诉我什么。在a different post中,同样的问题也得到了解决,但没有有效的解决办法。在Traceback (most recent call last):File "train.py", line 252, in main()File "train.py", line 231, in maintrain(net, training_dataset, targets,
我不明白这个错误在告诉我什么。在a different post中,同样的问题也得到了解决,但没有有效的解决办法。在Traceback (most recent call last):
File "train.py", line 252, in
main()
File "train.py", line 231, in main
train(net, training_dataset, targets, device, criterion, optimizer, epoch, args.epochs)
File "train.py", line 103, in train
loss.backward()
File "/home/hb119056/.local/lib/python3.6/site-packages/torch/tensor.py", line 107, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph)
File "/home/hb119056/.local/lib/python3.6/site-packages/torch/autograd/__init__.py", line 93, in backward
allow_unreachable=True) # allow_unreachable flag
RuntimeError: Function MulBackward0 returned an invalid gradient at index 0 - expected type torch.cuda.FloatTensor but got torch.FloatTensor
这是我代码中对应的段。在
^{pr2}$
非常感谢您的帮助,谢谢!在
欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
所有评论(0)