tensorflow 견적서가 보관하는 체크 포인트의 양을 제어하는 ​​방법은 무엇입니까?

Question

새로운 Estimator API가 교육 도중 검사 점을 자동으로 저장하고 교육이 중단되면 자동으로 마지막 검사 점에서 다시 시작된다는 사실을 알았습니다. 불행히도 마지막 5 개의 체크 포인트 만 유지하는 것 같습니다.

교육 중에 보관되는 검사 점 수를 제어하는 ​​방법을 알고 있습니까? ,

# 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로도 주문 가능합니다

Answer 1

Tensorflow tf.estimator.Estimator 다음과 같이 settings.You이 달성 할 수 런타임을 구성 할 수 tf.estimator.RunConfig 객체가 될 수있는 선택적 인수로 config 소요 LinearClassifier 등)이 estimator.Estimator까지 확장됩니다.