跑一个ocr的代码出现上述错误，解决方法：
1.设置batch_size和num_epochs能整除，即num_epochs/batch_size的值是一个正整数。
2.在DataLoader函数中，设置drop_last=True,如下图所示：
相关文章：
猜你喜欢
相关资源
相似解决方案
热门标签
跑一个ocr的代码出现上述错误，解决方法：
1.设置batch_size和num_epochs能整除，即num_epochs/batch_size的值是一个正整数。
2.在DataLoader函数中，设置drop_last=True,如下图所示：
相关文章：