web-dev-qa-db-ja.com

10エポックごとにモデルを保存するtensorflow.keras v2

私はテンソルフローv2のサブモジュールとして定義されたケラスを使用しています。 fit_generator()メソッドを使用してモデルをトレーニングしています。 10エポックごとにモデルを保存したい。どうすればこれを達成できますか?

Keras(tfのサブモジュールとしてではない)では、ModelCheckpoint(model_savepath,period=10)を指定できます。しかし、tf v2では、これをModelCheckpoint(model_savepath, save_freq)に変更しました。ここでsave_freq'Epoch'この場合、モデルはすべてのエポックに保存されます。 save_freqは整数であり、非常に多くのサンプルが処理された後でモデルが保存されます。しかし、私はそれが10エポックの後であることを望みます。どうすればこれを達成できますか?

10
Nagabhushan S N

tf.keras.callbacks.ModelCheckpointの使用save_freq='Epoch'を使用し、追加の引数period=10を渡します。

これは official docs で文書化されていませんが、それがその方法です(periodを渡すことができることが文書化されていることに注意してください。

3
bluesummers