解决Target size (torch.Size([1120])) must be the same as input size (torch.Size([1120,1]))

for t in range(10000):  # 迭代次数
    out = net(trainfeature_tensor)
#   print(out.size())
#   print(trainlabel_tensor.size())
    trainlabel_tensor.resize_(1120,1)
    loss = loss_func(out, trainlabel_tensor)
    print(t, loss)

    optimizer.zero_grad()
    loss.backward()
    optimizer.step()

print('Finished Training')

参考:(1条消息) 基于Pytorch的上市公司财务预警分析_chifengxin的专栏-CSDN博客

你可能感兴趣的:(新手问题,python,深度学习,pytorch)