Tensorflow поезд на сохраненной модели - PullRequest
0 голосов
/ 29 апреля 2020

Я использую тензорный поток для обучения РНН с более чем 200 КБ данных, учебный процесс занимает до 2 часов для каждой эпохи. Я сохраняю модель для каждой эпохи с пользовательским обратным вызовом и этой функцией: model.save_weights()

Могу ли я остановить тренировку и возобновить ее позже с прошлой эпохи, как это? Есть ли разница в результате? model.load_wieghts(last_epoch_dir) model.fit()

1 Ответ

2 голосов
/ 29 апреля 2020

Да, вы можете переучиться с последней эпохи, но проблема в том, что вы можете потерять состояние оптимизатора, но это не проблема, потому что оптимизатор вернется к своему первоначальному состоянию или даже лучше в течение нескольких эпох.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...