我在用botnet训练自己数据集的时候老是马上跑完一个epoch时出现RuntimeError，drop_last啦之类的都不管用

Question

我在用botnet训练自己数据集的时候老是马上跑完一个epoch时出现RuntimeError，在网上也找了很多方法，drop_last啦之类的都不管用，有人可以帮我解答一下吗