当前位置:网站首页>next(iter(dataloader))的一点点体会

next(iter(dataloader))的一点点体会

2022-06-26 17:42:00 LIZHEMGHONG

next(iter(dataloader))返回一个batch的数据,下面是我的一点代码截图,可以看出real_batch的长度和设置的batch_size一样。
此外dataloader的长度,即len(dataloader),就是总数据长度除以batchsize(向上取整)的大小 (这一点截图里没有体现,自己验证了的)
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

原网站

版权声明
本文为[LIZHEMGHONG]所创,转载请带上原文链接,感谢
https://blog.csdn.net/LIZHEMGHONG/article/details/114819026