RuntimeError:Function MulBackward0 returned an invalid gradient at index 0

问题:RuntimeError:Function MulBackward0 returned an invalid gradient at index 0 - expected type torch.FloatTensorbut got torch.cuda.FloatTensor

解决方法:关键是数据处理过程中格式发生了变化,而且找到开始变化的地方很麻烦,如果用print去看,很费时间,所以借用库函数:torchsnooper,可以看到每一行,每一个新增的变量的格式,类型等,很方便就能找到根本问题。

猜你喜欢

转载自blog.csdn.net/m0_46429066/article/details/104680493