除以batch_size后刚好余1,就发生了这个报错
数据加载器添加参数drop_last=True
,把最后一个批次丢弃
self.train_loader = DataLoader(train_set, batch_size=10, shuffle=True, num_workers=2, drop_last=True)
除以batch_size后刚好余1,就发生了这个报错
数据加载器添加参数drop_last=True
,把最后一个批次丢弃
self.train_loader = DataLoader(train_set, batch_size=10, shuffle=True, num_workers=2, drop_last=True)