Tensorflow: tf.estimator.Estimator保存唯一模型

checkpoint 模型保存數量默認爲5個， 爲方便管理，調整數量爲只保留1個

在train.py 中   加入

run_config = tf.estimator.RunConfig( save_checkpoints_secs=1e9, keep_checkpoint_max = 1 )

modle = tf.estimator.Estimator( model_fn=FLAGS.Model_function[0],   

                                                  model_dir=FLAGS.model_dir,

                                                  params=FLAGS.Model_fn_params,

                                                  config=run_config )

 

參考致謝：https://blog.csdn.net/roxxo/article/details/89886552