【解决方案】Expected object of type torch.FloatTensor but found type torch.DoubleTensor

Dear 丶 2022-04-17 06:29 305阅读 0赞

简述

在网上找了半天没有得到解答。
但是,在pycharm输入.之后,发现有一个函数很有意思,然后得到了解决

解决方案

假如报错的语句是

  • model是模型
  • input是输入(tensor的实体)

    model(input)

方法其实很简单

  1. mode(input.float())

发表评论

表情:
评论列表 (有 0 条评论,305人围观)

还没有评论,来说两句吧...

相关阅读