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

跑一个ocr的代码出现上述错误，解决方法：

1.设置batch_size和num_epochs能整除，即num_epochs/batch_size的值是一个正整数。
2.在DataLoader函数中，设置drop_last=True,如下图所示：