分类：tensor失败

2020
10-08

Pytorch 解决自定义子Module .cuda() tensor失败的问题

最近在刚从tensorflow转入pytorch，对于自定义的nn.Module碰到了个问题，即使把模组modle=Model().cuda()，里面的子Module的parameter都没有被放入cuda，导致输入是torch.cuda.FloatTensor，但是比如CNN的weight却还是torch.FloatTensor（当然最粗暴的方法就是直接在子模组里面都用了.cuda()但是问题并不在那，可能是调用子模组的时候，存在某些错误，导致父模组没有把子模组的parameter注册到pytorch中。而我遇到的错误就是，... 继续阅读 >