小吕同学真的不会敲代码 2022-05-19 20:44 采纳率: 57.1%
浏览 158
已结题

为什么每次训练完的模型的hdf5文件都没能保存下来呀?总是要重新训练

每次训练后,并没有训练好的模型文件hdf5存在指定位置中,是缺少什么保存模型的代码我没写么?新手上路,对比了很多训练模型的案例看不出自己少哪部分代码,求指点!保存记录loss的CSV文件也没能成功保存到文件夹下,很苦恼
【训练模型部分代码】

model_path = "bone_age_model_inception.hdf5"
checkpoint = ModelCheckpoint(model_path, monitor='val_loss', verbose=1, save_best_only=True, mode='min',
save_weights_only=True)
optim = optimizers.Nadam(lr=0.001, beta_1=0.9, beta_2=0.999, epsilon=1e-08, schedule_decay=0.0003)
reduceLROnPlat = ReduceLROnPlateau(monitor='val_loss', factor=0.8, patience=3, verbose=1, mode='auto', epsilon=0.0001,
cooldown=5, min_lr=0.0006)
early = EarlyStopping(monitor="val_loss", mode="min",
patience=10) # probably needs to be more patient, but kaggle time is limited
callbacks_list = [checkpoint, early, reduceLROnPlat]

bone_age_model_history = bone_age_model.fit_generator(generator=train_flow,
steps_per_epoch=STEP_SIZE_TRAIN,
validation_data=valid_flow,
validation_steps=STEP_SIZE_VALID,
epochs=EPOCHS,
callbacks=callbacks_list)

loss_history = bone_age_model_history.history['loss'] # 保存训练的loss
history_df = pd.DataFrame.from_dict(bone_age_model_history.history)
history_df.to_csv('./csv/incption_v3_loss_history1r.csv') # 这个csv文件也没有成功保存

print("Training complete !!!\n")

bone_age_model.save('bone_age_model_inception.hdf5') # 保存HDF5模型文件

bone_age_model.load_weights("bone_age_model_inception.hdf5") #加载模型
test_X, test_Y = next(test_flow)
pred_Y = mu + sigma * bone_age_model.predict(x=test_X, batch_size=16, verbose=1)

运行结果及报错内容

运行都正常,没报错,就是hdf5不知道怎么保存到目录里,明明也用了save函数了

我的解答思路和尝试过的方法

尝试过把model_path换成绝对路径,还是无济于事

  • 写回答

4条回答 默认 最新

  • 狗小那条花 2022-05-19 21:05
    关注
    获得1.05元问题酬金

    save模型之后,再用要load。。

    评论

报告相同问题?

问题事件

  • 系统已结题 5月27日
  • 修改了问题 5月20日
  • 修改了问题 5月20日
  • 赞助了问题酬金5元 5月19日
  • 展开全部

悬赏问题

  • ¥30 这是哪个作者做的宝宝起名网站
  • ¥60 版本过低apk如何修改可以兼容新的安卓系统
  • ¥25 由IPR导致的DRIVER_POWER_STATE_FAILURE蓝屏
  • ¥50 有数据,怎么建立模型求影响全要素生产率的因素
  • ¥50 有数据,怎么用matlab求全要素生产率
  • ¥15 TI的insta-spin例程
  • ¥15 完成下列问题完成下列问题
  • ¥15 C#算法问题, 不知道怎么处理这个数据的转换
  • ¥15 YoloV5 第三方库的版本对照问题
  • ¥15 请完成下列相关问题!