神经网络训练中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 

个人理解,如有错误,欢迎指正。

上一篇 下一篇

猜你喜欢

热点阅读