pytorch 参数注册问题
Posted yutingmoran
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了pytorch 参数注册问题相关的知识，希望对你有一定的参考价值。
在实现deepfm是进行特征编码时遇到RuntimeError: Function AddBackward0 returned an invalid gradient at index 1 - expected type torch.FloatTensor but got torch.cuda.FloatTensor问题，但模型和输入都已经to(device),经检查发现nn.ModuleList nn.ModuleDict的参数无法通过model.to(device)自动注册,要手动注册。
以上是关于pytorch 参数注册问题的主要内容，如果未能解决你的问题，请参考以下文章
Pytorch中 类Parameter的解析，类内成员函数.parameters()的源码分析，参数集合的获取，参数的注册赋值源码分析