问题描述:在生成dataloader的时候,训练集可以跑,测试集出现此错误:RuntimeError: stack expects each tensor to be equal size, but got [200] at entry 0 and [116] at entry 1。
解决思路:在生成dataloader需要生成dataset,那么我的错误发生原因是因为dataset中有一个minibatch的数据个数与其他minibatch不同,遂进入自定义dataset方法中查看,通过打印调试,发现是数据集label的问题。
解决方案:进入dataset中,打印dataset输出的结果。