这个屌丝很懒，什么也没留下！

【BUG解决】RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor)_inputs = inputs.to('cuda:0')

作者：繁依Fanyi0 | 2024-02-18 01:31:47

inputs = inputs.to('cuda:0')

RuntimeError: Input type (torch.FloatTensor) and weight type
(torch.cuda.FloatTensor) should be the same

device = torch.device('cuda:0')
inputs = inputs.to(device)
1
2

inputs = inputs.cuda()
1

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/103319