2017-02-28 2 views
0

Quelle est la meilleure façon de désactiver l'évaluation si j'utilise tf.learn.Experiment?Désactiver eval en utilisant tf.learn.Experiment

Je cours ce model qui construit une expérience.

tf.contrib.learn.Experiment(
     estimator=estimator, 
     train_input_fn=train_input_fn, 
     train_steps=FLAGS.num_train_steps, 
     train_monitors=[export_monitor], 
     eval_input_fn=eval_input_fn, 
     eval_steps=FLAGS.num_eval_steps, 
     eval_metrics=_create_evaluation_metrics(), 
     min_eval_frequency=100) 

Pour déboguer un problème avec l'évaluation, je souhaite empêcher l'exécution de l'évaluation. Y a-t-il un moyen facile de faire ceci?

Répondre

2

La réponse dépend de la méthode que vous invoquez sur Experiment. Vous allez probablement appeler train_and_evaluate, par exemple, si le type de tâche de TF_CONFIG est défini sur "master" (voir this code).

Dans ce cas, vous aurez envie de mettre min_eval_frequency-0 ou None (cf this code)

+0

Merci, mais il semble que cela désactiverait l'évaluation périodique au cours de la formation, mais il semble que l'évaluation serait encore une fois courir après la formation complète. –

+0

alors utilisez simplement la méthode train() – rhaertel80