PyTorch-Error:-Expected-more-than-1-value-per-channel-when-training

很简单

使用 DataLoader 增加一个参数 drop_last=True

如

self.train_loader = DataLoader(self.train_set,
                               batch_size=self.batch_size,
                               shuffle=True, 
                               pin_memory=False,
                               num_workers=self.num_workers,
                               collate_fn=collater,
                               drop_last=True
                              )