很简单
使用 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
)