【torch错误】Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
使用torch训练模型时,出现错误:
“Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same”
错误原因:
模型放入GPU后也只能处理数据类型为cuda的GPU数据,但你在把数据输入进模型之前并未做转换,依然是张量类型 。因此出错,二者必须一致。
例如:
device = torch.device('cuda:0')
Mymodel = VGGModel().to(device)
原创
2022-02-17 21:28:48 ·
1947 阅读 ·
0 评论