next(iter(dataloader))返回一个batch的数据,下面是我的一点代码截图,可以看出real_batch的长度和设置的batch_size一样。
此外dataloader的长度,即len(dataloader),就是总数据长度除以batchsize(向上取整)的大小 (这一点截图里没有体现,自己验证了的)
next(iter(dataloader))的一点点体会
最新推荐文章于 2023-09-27 14:42:01 发布
next(iter(dataloader))返回一个batch的数据,下面是我的一点代码截图,可以看出real_batch的长度和设置的batch_size一样。
此外dataloader的长度,即len(dataloader),就是总数据长度除以batchsize(向上取整)的大小 (这一点截图里没有体现,自己验证了的)