我不明白这个错误在告诉我什么。在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}$

非常感谢您的帮助,谢谢!在

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐