神经网络训练中epoch, iteration, batch的关
2020-05-09 本文已影响0人
不太聪明的亚子
epoch:整个数据集被拿来训练的次数
batch:确定了epoch后,每次从数据集中取batchsize个数据来完成一次模型训练,参数更新
iteration:完成一次batch的训练就是一个iteration
假设 :
数据集:DataSet,包含1000个样本
那么:
L = len(DataSet) = 1000
若epoch = 1,batch = 100,iteration = L*epoch/batch = 10
若epoch = 10, batch = 100, iteration = L*epoch/batch = 100
个人理解,如有错误,欢迎指正。