小编Hap*_*orm的帖子

fit_generator中"samples_per_epoch"和"steps_per_epoch"之间的区别是什么

我好几天都对这个问题感到困惑......

我的问题是,为什么训练时间与我的生成器的batch_size设置为"1"和"20"之间存在巨大差异.

如果我设置的batch_size1时,训练时间1个时期为约180〜200秒.如果我设置的batch_size20时,训练时间1个时期为约3000〜3200秒.

然而,这些训练时间之间的这种可怕差异似乎是异常的...,因为它应该是相反的结果:batch_size = 1,训练时间 - > 3000~3200秒.batch_size = 20,训练时间 - > 180~200秒.

我的生成器的输入不是文件路径,而是通过调用"np.load()"已经加载到内存中的numpy数组.所以我认为I/O权衡问题不存在.

我正在使用Keras-2.0.3,我的后端是tensorflow-gpu 1.0.1

我已经看到了这个合并PR的更新,但似乎这个改变不会影响任何东西.(用法与原版相同)

这里的链接是我自定义生成器的要点和fit_generator的一部分.

有人可以帮我解释一下这个问题吗?非常感谢:)

keras

25
推荐指数
1
解决办法
3万
查看次数

标签 统计

keras ×1