如何以最佳權重保存 tensorflow model

Question

我正在訓練一個 LSTM，並且對於 300 個 epoch 訓練序列期間的部分 epoch（范圍不一致），它在很長一段時間內表現出非常好的准確性。 我希望此時能夠保存整個 model，但除了使用檢查點然后恢復並保存新的 model 之外，似乎找不到其他方法。

有沒有辦法我可以使用 model.save 僅在 model 具有最佳權重時捕獲它？

Answer 1

使用save _best_only參數：

model_checkpoint_callback = tf.keras.callbacks.ModelCheckpoint(
    filepath=checkpoint_filepath,
    save_weights_only=True,
    monitor='val_accuracy',
    mode='max',
    save_best_only=True)