我正在将tf.estimator.Estimator与这样的代码一起使用
run_config = tf.estimator.RunConfig(
model_dir=output_dir,
save_summary_steps=None,
save_checkpoints_steps=FLAGS.save_checkpoints_steps,
save_checkpoints_secs=None
)
estimator = tf.estimator.Estimator(
model_fn=model_fn,
config=run_config
)
# prepare training data
# ...
estimator.train()
# prepare eval data
# ...
estimator.evaluate()
似乎估算器会自动保存几个文件:
- events.out.tfevents*
- graph.pbtxt
- etc.
我的模型很大,并且包含大(常数)矩阵,即使将save_summary_steps
设置为None
,也使提到的两个文件非常大。
每个培训过程都是从将这些大文件写入磁盘开始的,并且要花费大量时间。
我不希望这种自动保存,尤其是events.out.tfevents*
和graph.pbtxt
。有什么解决办法吗?
我进行了一些搜索,发现将save_checkpoints_steps
和save_checkpoints_secs
都设置为None
时将禁用检查点保存(包括graph.pbtxt),但是接下来步骤estimator.evaluate()
不起作用。