1.F.cross_entropy(outputs, labels) RuntimeError: Dimension out of range (expected to be in range of [-1, 0], but got 1)
参考:
- https://junchu.blog.csdn.net/article/details/106825660
- https://blog.csdn.net/weixin_41068770/article/details/886342932
2.RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #4 'mat1'