在實現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)自動注冊,要手動注冊。
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。