当前位置: 代码迷 >> 综合 >> 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)

热度:41   发布时间:2024-02-20 23:53:47.0

在这里插入图片描述跑一个ocr的代码出现上述错误,解决方法:

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

  相关解决方案