Загрузка контрольной точки из обученной модели с использованием оценщика

Question

Я хочу сделать очень простое задание. Давайте предположим, что я выполнил модель и сохранил несколько контрольных точек и метад для этой модели, используя tf.estimator. Мы можем снова предположить, что у меня есть 3 контрольных пункта. 1, 2 и 3. Пока я оцениваю результаты тренировок на тензорной доске, я понимаю, что контрольная точка 2 обеспечивает лучшие веса для моей цели.

Поэтому я хочу загрузить контрольную точку 2 и сделать свои прогнозы. Я просто хочу спросить: можно ли удалить контрольную точку 3 из модельного каталога и позволить оценщику автоматически загрузить ее из контрольной точки 2, или я могу что-то сделать, чтобы загрузить конкретную контрольную точку для прогнозов .my?

Спасибо.

Answer 1

Да, Вы можете.По умолчанию Estimator загрузит последнюю доступную контрольную точку в model_dir.Таким образом, вы можете либо удалить файлы вручную, либо указать файл контрольной точки с помощью

warm_start = tf.estimator.WarmStartSettings(ckpt_to_initialize_from='file.ckpt')

и передать его в оценщик

tf.estimator.Estimator(model_fn=model_fn,
                       config=run_config,
                       model_dir='dir',
                       warm_start_from=warm_start)

Последний вариант не будет путать сводки по тензорной доске, поэтому он обычно чище