Expected tensor for argument #1 ‘input‘ to have the same device as tensor for argument #2 ‘weight‘;
生活随笔
收集整理的這篇文章主要介紹了
Expected tensor for argument #1 ‘input‘ to have the same device as tensor for argument #2 ‘weight‘;
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
這種情況出現的原因之一是創建網絡的語句沒有放進train函數里
比如主文件中有train(),test(),eval()......等各個函數,但是創建網絡的語句:net=createnet()放在各個函數外面,然后在train(),test(),eval()這幾個函數里面使用函數外面創建的net,此時就會報錯
解決方法:
把創建net的語句分別放進各個函數里
總結
以上是生活随笔為你收集整理的Expected tensor for argument #1 ‘input‘ to have the same device as tensor for argument #2 ‘weight‘;的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 删除计算机文件的几种方法,电脑删除不了文
- 下一篇: python文件读写方法手机_pytho