batch.size是什么
时间: 2024-06-07 22:03:08 浏览: 134
保存数据-pytorch dataloader 取batch_size时候出现bug的解决方式
在深度学习中,batch size指的是每次训练时,同时输入神经网络的样本数。通常情况下,我们无法将整个数据集一次性输入到神经网络中进行训练,因此需要将数据分成若干批,每批包含一定数量的样本,这个数量就是batch size。在训练过程中,我们会对每一批样本进行前向传播、反向传播以及参数更新,然后再取下一批样本进行训练,直到整个数据集都被用于训练为止。
阅读全文