alex_bn_lee

导航

< 2025年3月 >
23 24 25 26 27 28 1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28 29
30 31 1 2 3 4 5

统计

【597】keras 回调(callbacks)函数 以及 模型保存

参考:Tensorflow回调(callbacks)函数的使用方法

参考:Keras中的回调函数Callbacks详解

参考:keras保存模型中的save()和save_weights()

参考:关于 keras.callbacks设置模型保存策略


  实例化一个 myCallback 对象 callbacks

  在 model.fit() 函数中添加 callbacks 参数

  model.save() :保存整个模型结构与参数:保存整个模型结构与参数

  model.save_weights() :仅仅保存模型的参数:仅仅保存模型的参数

  model.load_model() :加载存储的信息

1
2
3
4
5
callbacks = [
    keras.callbacks.ModelCheckpoint("oxford_segmentation.h5", save_best_only=True)
]
 
model.fit(train_gen, epochs=epochs, validation_data=val_gen, callbacks=callbacks)

  默认是每一次poch,但是这样硬盘空间很快就会被耗光.

  将save_best_only 设置为True使其只保存最好的模型,值得一提的是其记录的acc是来自于一个monitor_op,其默认为"val_loss",其实现是取self.best为 -np.Inf. 所以,第一次的训练结果总是被保存.

  • Callbacks:回调函数是一个函数的合集,会在训练的阶段中所使用。你可以使用回调函数来查看训练模型的内在状态和统计。你可以传递一个列表的回调函数(作为 callbacks 关键字参数)到 Sequential 或 Model 类型的 .fit() 方法。在训练时,相应的回调函数的方法就会被在各自的阶段被调用。

posted on   McDelfino  阅读(313)  评论(0编辑  收藏  举报

编辑推荐:
· AI与.NET技术实操系列(二):开始使用ML.NET
· 记一次.NET内存居高不下排查解决与启示
· 探究高空视频全景AR技术的实现原理
· 理解Rust引用及其生命周期标识(上)
· 浏览器原生「磁吸」效果!Anchor Positioning 锚点定位神器解析
阅读排行:
· DeepSeek 开源周回顾「GitHub 热点速览」
· 记一次.NET内存居高不下排查解决与启示
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布:重大改进与新特性概览!
· .NET10 - 预览版1新功能体验(一)
历史上的今天:
2012-07-05 【055】长江水文数据自动记录程序
点击右上角即可分享
微信分享提示