https://www.jianshu.com/p/0be7a375bdbe
https://blog.csdn.net/qq_38410428/article/details/82973895
計算中有的參數為cuda型,有的參數卻是cpu型,就會遇到這樣的錯誤。
解決辦法:
該加.cuda()的加上,不該用.cpu()的地方去掉它。
https://www.jianshu.com/p/0be7a375bdbe
https://blog.csdn.net/qq_38410428/article/details/82973895
計算中有的參數為cuda型,有的參數卻是cpu型,就會遇到這樣的錯誤。
解決辦法:
該加.cuda()的加上,不該用.cpu()的地方去掉它。
本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。