平凡之上
博观而约取,厚积而薄发! 守得云开见月明。

#P14-P15

 

torchvision中数据集的使用

 

DataLoader的使用

batch_size:每批次抓取的数据个数。drop_last:如果DataSet大小未被批处理大小可分离,则将其设置为true以删除最后的不完整批次。如果DataSet的尺寸不可分可于批次大小,则最后批处理将更小。(默认:FALSE)

posted on 2021-11-19 11:28  平凡之上  阅读(17)  评论(0编辑  收藏  举报