python - keras に保存されたモデルを使用して CNN のトレーニングを続行する

Question

エポック数が 25 の keras ライブラリを使用して CNN モデルをトレーニングします。最初に 10 エポックでモデルを実行し、次のコード行でモデルを保存できますか。

model.fit_generator(training_set,
                     steps_per_epoch = 100000,
                     epochs = 10,
                     validation_data = test_set,
                     validation_steps = 40000)
from keras.models import load_model
model.save('my_model.h5') 

次に、Python を再起動し、次の 15 エポックを次のコードのように同じデータセットで実行し続けます。

model = load_model('my_model.h5')
model.fit_generator(training_set,
                     steps_per_epoch = 100000,
                     epochs = 15,
                     validation_data = test_set,
                     validation_steps = 40000)

トレーニングを続けるだけで十分ですか？または、仕事を続けるために他のステップを実行する必要があります。どんなサポートでも大歓迎です。

Answer 1

はい、これでmodel.save問題ありません。重み、モデル アーキテクチャ、オプティマイザーの状態が保存されるので、問題なくトレーニングを再開できます。