Pytorch 报错:RuntimeError: invalid argument 0: Tensors must have same number of dimensions: got 2 and
Pytorch 报错:RuntimeError: invalid argument 0: Tensors must have same number of dimensions: got 2 and 1 at /pytorch/aten/src/THC/generic/THCTensorMath.cu:62这种错误有两种可能:你输入的图像数据的维度不完全是一样的 ,比如是训练的数据有100组...
转载
2020-04-08 21:36:49 ·
16554 阅读 ·
5 评论