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.Mo...
原创
2019-11-25 09:40:33 ·
1639 阅读 ·
1 评论