可以在定义dataloader时将drop_last参数设置为True,这样最后一个batch如果数据不足时就会被舍弃,而不会报错。例如:
dataloader = torch.utils.data.DataLoader(dataset, batch_size=batch_size, drop_last=True)
另外,也可以在数据集的 len 函数中返回整除batch_size的长度来避免最后一个batch报错。
可以在定义dataloader时将drop_last参数设置为True,这样最后一个batch如果数据不足时就会被舍弃,而不会报错。例如:
dataloader = torch.utils.data.DataLoader(dataset, batch_size=batch_size, drop_last=True)
另外,也可以在数据集的 len 函数中返回整除batch_size的长度来避免最后一个batch报错。