在每个epoch后保存模型权重
filepath可以包含命名格式化选项,可以由epoch的值和logs的键(由on_epoch_end参数传递)来填充。
例如:如果filepath是weights.{epoch:02d}-{val_loss:.2f}.hdf5,则模型检查点将与文件名中的epoch号和验证损失一起保存。
参数 | 描述 |
---|---|
filepath | 保存模型文件的路径。 |
monitor | 要监测的直白哦 |
verbose | 详细模式 |
save_best_only | 保存最好的权重 |
mode | |
save_weights_only | True:只保存权重 |
period |
init
__init__(
filepath,
monitor='val_loss',
verbose=0,
save_best_only=False,
save_weights_only=False,
mode='auto',
period=1
)
参考:
官网
https://www.w3cschool.cn/tensorflow_python/tf_keras_callbacks_ModelCheckpoint.html