通常 dataloader 中可设置 batch size,dataloader会将图像进行一个打包成tensor加快训练。
假设训练集有8张图像,batch_size=4,epoch=2,那么每次训练,dataloader迭代器将会送1个tensor(4张图像)的数据,送2次即完成了整个训练集的遍历,即1个epoch。
关于batch size/epoch/dataloader
最新推荐文章于 2024-03-19 21:02:38 发布
本文解释了在深度学习中,如何通过设置dataloader的batch_size来优化训练过程,如将8张图像打包成一个包含4张的tensor,每个epoch迭代两次完成一个训练集的遍历。
摘要由CSDN通过智能技术生成