禁用保存events.out.tfevents和graph.pbtxt tensorflow.Estimator

时间:2019-05-07 17:10:40

标签: tensorflow tensorflow-estimator

我正在将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_stepssave_checkpoints_secs都设置为None时将禁用检查点保存(包括graph.pbtxt),但是接下来步骤estimator.evaluate()不起作用。

0 个答案:

没有答案