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

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

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

Спасибо.


machine-learning tensorflow deep-learning tensorflow-estimator
person TheWho    schedule 15.09.2018    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Да, ты можешь. По умолчанию 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)

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

person Sharky    schedule 21.02.2019