fit_generator是否在每个时期加载不同的数据集?

时间:2018-03-19 00:49:36

标签: tensorflow machine-learning deep-learning keras

据我所知,一个纪元通常表示每个训练样本的一次迭代。 Keras fit_generator函数获得了参数sample_per_epoch。让我们说我有20000个训练样本,我决定将samples_per_epoch设置为5000.下一个时代会发生什么?它会选择相同的5000个训练样本集,还是从新的随机集开始?

1 个答案:

答案 0 :(得分:0)

如果您尝试使用相同的模型对象,那么当您在同一个会话中时,我将从5001开始,否则它将再次从第1纪元开始。 会话完成后,模型对象将过期。为避免此类问题,您可以保存模型以便可以使用它,它将从5001年开始。它适用于相同的数据集