私はテンソルフローv2のサブモジュールとして定義されたケラスを使用しています。fit_generator()
メソッドを使用してモデルをトレーニングしています。10エポックごとにモデルを保存したい。どうすればこれを達成できますか?
Keras(tfのサブモジュールではない)では、を指定できますModelCheckpoint(model_savepath,period=10)
。しかし、TF V2に、彼らはこれを変更したModelCheckpoint(model_savepath, save_freq)
場所save_freq
することができ'epoch'
、その場合、モデルはすべてのエポックを保存されます。save_freq
が整数の場合、非常に多くのサンプルが処理された後でモデルが保存されます。しかし、私はそれが10エポックの後であることを望みます。どうすればこれを達成できますか?
WARNING:tensorflow:'period' argument is deprecated. Please use 'save_freq' to specify the frequency in number of samples seen.
つまり、この機能はまもなく終了します。その場合、どうすればこれを達成できますか?