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')
解决Target size (torch.Size([1120])) must be the same as input size (torch.Size([1120,1]))
最新推荐文章于 2023-11-07 10:38:46 发布