在很多情况下,当我们处理数据时,我们希望将数据分成批次进行处理。在PyTorch中,可以使用DataLoader来完成这个任务。DataLoader中有一个参数叫做drop_last,它的默认值是False,表示如果最后一个batch的数据不足一个batch_size大小,则保留该batch。如果将drop_last设置为True,则最后一个batch的数据将被舍弃,不足一个batch_size大小的数据将被丢弃。
drop_last设置为True
最新推荐文章于 2024-06-15 06:22:58 发布