发新帖

关于keras中断后继续训练initial_epoch问题

[复制链接]
528 0

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
本帖最后由 InsightDev 于 2018-5-17 12:13 编辑

因为learning_rate随epoch变化,所以中断后继续训练想从指定的epoch开始,是否可以从上次训练的中间结果获取initial_epoch.

initial_epoch见mModel.fit_generator参数列表。

  1. callbacks = [
  2.      LearningRateScheduler(scheduler_32k_48k_64k),
  3.      ModelCheckpoint(weights_path, monitor='val_loss', verbose=1, mode='min'),
  4. ];

  5. datagen = ImageDataGenerator(
  6.      horizontal_flip=True,
  7.       width_shift_range=0.125,
  8.      height_shift_range=0.125,
  9.      fill_mode='constant', cval=0.0);

  10. datagen.fit(x_train);

  11. mModel.fit_generator(datagen.flow(x_train, y_train, batch_size),
  12.            epochs=epochs, callbacks=callbacks,
  13.            validation_data=(x_valid, y_valid), initial_epoch=0);
复制代码

我知道答案 回答被采纳将会获得10 金币 + 10 金币 已有0人回答
本楼点评(1) 收起
  • yunhai_luo如果你的scheduler_32k_48k_64k还跟上一代的学习率有关,那么还需要确保模型状态(特别是学习率)也被正确导入。
    2018-5-17 13:42 回复
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

快速回复 返回顶部 返回列表