Как использовать fit_generator при загрузке модели в Keras?

Question

Я обучил двоичный классификатор, затем сохранил модель в виде файла .h5.

Первоначально я использовал fit_generator для его обучения.

model.fit_generator(
train_generator,
class_weight=class_weights,
steps_per_epoch=nb_train_samples // batch_size,
epochs=epochs,
validation_data=validation_generator,
validation_steps=nb_validation_samples // batch_size,
callbacks=my_callbacks
)

Когда я вызываю load_model и загружаю его снова, чтобы продолжить обучение там, где я остановился, использую ли я точно такой же код для fit_generator, ссылаясь на старый train_generator и validation_generator?

Документация Keras очень скудна по этому вопросу, и практически нет примеров использования load_model с fit_generator вместе в сети.

Answer 1

Да. Почему бы тебе? Нагрузка будет загружать веса. Генератор будет генерировать новые обучающие образцы. Конечно, он может не начинаться точно там, где он закончился, но до тех пор, пока перетасовывается, все будет в порядке.