RuntimeError: The size of tensor a (8) must match the size of tensor b (7) at non-singleton dimensio
在出现类似问题报错,并且该错误出现的情景为前面训练正常,在当前训练的epoch快要结束时出现报错,则该错误原因为在训练过程中所用的数据集数量与所设置的batchsize可能不是整除的关系,导致出错。解决方法,在DataLoader函数中加上一个参数drop_last=False,具体示例如下:dataloader=DataLoader(ImageDataset(opt.dataroot,trans