私はテンソルフローv2のサブモジュールとして定義されたケラスを使用しています。 fit_generator()
メソッドを使用してモデルをトレーニングしています。 10エポックごとにモデルを保存したい。どうすればこれを達成できますか?
Keras(tfのサブモジュールとしてではない)では、ModelCheckpoint(model_savepath,period=10)
を指定できます。しかし、tf v2では、これをModelCheckpoint(model_savepath, save_freq)
に変更しました。ここでsave_freq
は'Epoch'
この場合、モデルはすべてのエポックに保存されます。 save_freq
は整数であり、非常に多くのサンプルが処理された後でモデルが保存されます。しかし、私はそれが10エポックの後であることを望みます。どうすればこれを達成できますか?
tf.keras.callbacks.ModelCheckpoint
の使用save_freq='Epoch'
を使用し、追加の引数period=10
を渡します。
これは official docs で文書化されていませんが、それがその方法です(period
を渡すことができることが文書化されていることに注意してください。