【解决方案】Expected object of type torch.FloatTensor but found type torch.DoubleTensor
生活随笔
收集整理的這篇文章主要介紹了
【解决方案】Expected object of type torch.FloatTensor but found type torch.DoubleTensor
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
簡述
在網上找了半天沒有得到解答。
但是,在pycharm輸入.之后,發現有一個函數很有意思,然后得到了解決
解決方案
假如報錯的語句是
- model是模型
- input是輸入(tensor的實體)
方法其實很簡單
mode(input.float())總結
以上是生活随笔為你收集整理的【解决方案】Expected object of type torch.FloatTensor but found type torch.DoubleTensor的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 风格迁移理论
- 下一篇: 【Kaggle-MNIST之路】两层的神