记录报错

错误1:

Expected object of backend CPU but got backend CUDA for argument #2 'weight'   

#之前:
    model = model.cuda()
#之后:
    model = torch.nn.DataParallel(model).cuda()

错误2:

IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)

 #之前:
# 前向传播计算输出
            outputs = model(input)
            label = label.squeeze()
            outputs.detach_()
            # 计算准确率,记录损失
            loss = criterion(outputs[i], label)
#之后:
             loss = criterion(outputs, label)            
#之前label有一个维度问题 
label.size()=tensor([16,1])

label = label.squeeze()

label.size()=tensor([16])
原文地址:https://www.cnblogs.com/Manuel/p/13267595.html