Продолжить тренировку CNN с сохраненной моделью в Керасе

Question

Я тренирую модель CNN с библиотекой keras с номерами эпох, равными 25. Можно ли запустить модель в первый раз с 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 сохраняет веса, архитектуру модели и состояние оптимизатора, чтобы вы могли без проблем возобновить обучение.