在出现类似问题报错，并且该错误出现的情景为前面训练正常，在当前训练的epoch快要结束时出现报错，则该错误原因为在训练过程中所用的数据集数量与所设置的batchsize可能不是整除的关系，导致出错。

解决方法，在DataLoader函数中加上一个参数drop_last=False，具体示例如下：

dataloader = DataLoader(ImageDataset(opt.dataroot, transforms_=transforms_, unaligned=True),
                        batch_size=opt.batchSize, shuffle=True, num_workers=opt.n_cpu, drop_last=False)