Keras2.0 fit_generator 方法签名改变导致1.0版本代码训练变慢

发表于 | 更新于 | 分类于

Keras2.0 fit_generator 方法签名改变导致1.0版本代码训练变慢

keras2.0 版本fit_generator方法签名

1
2
3
4
5
6
model.fit_generator(
    train_generator,
    steps_per_epoch=nb_train_samples //  batch_size,
    validation_data=validation_generator,
    validation_steps = nb_validation_samples // batch_size,
    epochs=epochs)

keras1.0 版本fit_generator方法签名

1
2
3
4
5
6
model.fit_generator(
       train_generator,
       samples_per_epoch=nb_train_samples,
       epochs=epochs,
       validation_data=validation_generator,
       nb_val_samples=nb_validation_samples)

虽然1.0版本的代码在2.0版本上可以运行，但是因为参数没有正确设置，训练期间跑验证集时速度很慢（validation_steps的值为1），我自己测试时一轮需要的时间多4倍。
其实当这个函数在执行时，Keras会提示开发者对参数进行修改，所以使用Keras时千万不要忽略了提示信息。