RuntimeError: Expected object of scalar type Double but got scalar type Float for argument #2 'mat2'

偏执的太偏执、 2024-04-17 05:38 135阅读 0赞

问题:RuntimeError: Expected object of scalar type Double but got scalar type Float for argument #2 ‘mat2’

问题代码段:

  1. embedding1 = network.forward(train_support)

解决方法:

  1. train_support = train_support.clone().detach().float().cuda()

发表评论

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

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

相关阅读