记录一个在使用 Pytorch 0.4.0 过程中遇到的 RuntimeError
最近在使用新版 Pytorch 0.4.0(听说1.0版本马上就要出了…)训练 GAN 的时候遇到了这样一个BUG,RuntimeError: Expected object of type torch.DoubleTensor but found type torch.cuda.FloatTensor for argument #2 ‘weight’ 。相信很多初学者都会在刚开始的时候遇到这样的...
原创
2018-05-04 15:36:36 ·
17169 阅读 ·
6 评论