next(iter(dataloader))返回一个batch的数据,下面是我的一点代码截图,可以看出real_batch的长度和设置的batch_size一样。
此外dataloader的长度,即len(dataloader),就是总数据长度除以batchsize(向上取整)的大小 (这一点截图里没有体现,自己验证了的)
![]()
![]()
版权声明:本文为LIZHEMGHONG原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。
next(iter(dataloader))返回一个batch的数据,下面是我的一点代码截图,可以看出real_batch的长度和设置的batch_size一样。
此外dataloader的长度,即len(dataloader),就是总数据长度除以batchsize(向上取整)的大小 (这一点截图里没有体现,自己验证了的)
![]()
![]()