テンソルフロー推定器によって保持されるチェックポイントの量を制御する方法は？

Question

私は、トレーニング中に新しいEstimator APIが自動的にチェックポイントを保存し、トレーニングが中断されたときに最後のチェックポイントから自動的に再起動することに気付きました。残念ながら、最後の5つのチェックポイントしか保持していないようです。

トレーニング中に保持されるチェックポイントの数を制御する方法を知っていますか？

Answer 1

Tensorflow tf.estimator.Estimatorは、次のようにsettings.Youはこれを達成することができますランタイムを設定するには、tf.estimator.RunConfigオブジェクトにすることができ、オプションの引数としてconfigを取ります

# Change maximum number checkpoints to 25 
run_config = tf.estimator.RunConfig() 
run_config = run_config.replace(keep_checkpoint_max=25) 

# Build your estimator 
estimator = tf.estimator.Estimator(model_fn, 
            model_dir=job_dir, 
            config=run_config, 
            params=None) 

configパラメータは、すべてのクラス（DNNClassifier、DNNLinearCombinedClassifierで提供されていますLinearClassifierなど）が含まれており、これはestimator.Estimatorです。