tensorflow - 如何使用 TensorFlow 的 Keras API 为每个 epoch 的保存模型生成唯一名称
问题描述
我正在训练模型fit_generator()
并希望为每个时代保存的 wights 生成唯一的名称
已经尝试过:查看后面的代码
代码:
model_path = '.\checkpoints\cp{}.ckpt'.format(time())
cp_callback = tf.keras.callbacks.ModelCheckpoint(model_path,
verbose=1,
period=2)
model.fit_generator(..........,callbacks=[cp_callback])
预期:生成唯一的检查点名称
,例如 epoch_4.ckpt 或 epoch_5.ckpt
实际:每次保存时,覆盖现有检查点
解决方案
您可以尝试将 epoch 变量添加到文件路径变量
filepath = ".\checkpoints\cp-{epoch:02d}.hdf5"
cp_callback = tf.keras.callbacks.ModelCheckpoint(filepath, verbose=1, period=2)
这里的 Keras 文档已经提到了这一点。
推荐阅读
- vue.js - 在 nuxt js 中使用 axios 验证过期后重定向
- c# - 创建自定义渲染器时出现 TypeLoadException
- amazon-web-services - KMS 和 S3 存储桶
- c# - Azure 函数不运行异步方法
- wpf - 从资源字典中选择图像源路径
- python - 在 Python 中使用 csv 阅读器的变量范围
- tensorflow - 如何在谷歌colab中导入张量流
- datatable - 如何处理由于SPSS中每个月剩余的相同案例而重复的数据
- c - 如何定义 functoin poiner(__fastcall) 在 linux gcc 中调用?
- html - 结合表单使用jquery加载功能