Как использовать tennorflow.estimator.Estimator latest_checkpoint как часть параметров для model_fn? - PullRequest
0 голосов
/ 22 марта 2020

У меня есть модель, которая использует номер итерации в качестве гиперпараметра. Чтобы восстановить тренировочный процесс, в котором я ушел, я хотел бы использовать номер итерации с последней контрольной точки.

classifier = tf.estimator.Estimator(
        model_fn=model_fn,
        params=params,
        config=run_config)

, но для получения последней контрольной точки мне нужно сделать

latest_checkpoint = classifier.latest_checkpoint()

Есть ли способ отправить latest_chekpoint на мой model_fn?

...