义子

如何让Pytorch中自定义子Module的.tensor().cuda()操作成功执行？

本文共计828个文字，预计阅读时间需要4分钟。最近从TensorFlow转入PyTorch，在自定义nn.Module时遇到了问题。将模型+model=Model().cuda()后，子Module的parameter都没有被放入cuda，