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

IndexError:Dimension out of range(expected to be in range of [-1,0],but got 1)_第1张图片跑一个ocr的代码出现上述错误,解决方法:

1.设置batch_size和num_epochs能整除,即num_epochs/batch_size的值是一个正整数。
2.在DataLoader函数中,设置drop_last=True,如下图所示:
在这里插入图片描述

你可能感兴趣的:(IndexError错误,深度学习)