gimme5的博客

原创 错误记录：Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

根据我的理解，只有模型中被正确注册的模块 （properly registered）可以被正确的初始化到想要的设备上，而一般的list类型并不属于pytorch中的模块，所以并不算是被正确注册了。可以看到所有的自定义的模块都在__init__函数进行初始化了，似乎这个也不是问题所在。那么到底是哪里的板块没有被成功初始化呢？打印出来似乎也没什么问题。但是这里又说是模型参数和输入不在同一设备上，那么应该就是自定义模型的问题。