如何让tf.contrib.learn.estimator不添加摘要?

时间:2017-02-17 08:32:19

标签: python tensorflow

我正在使用tf.contrib.learn.DNNClaisifer来训练我的模型,同时估算器会在model_dir中生成张量流摘要,这会花费大量空间。我没有透露摘要,因为我找不到任何参数或接口来指定是否写入tensorflow摘要。你能告诉我如何在tf.contrib.learn.DNNClasifier中对摘要编写器进行diable吗?

1 个答案:

答案 0 :(得分:1)

似乎没有这样的选项,但您可以从code禁用它。删除contrib.learn.python.learn.estimators.dnn._add_hidden_layer_summary

的内容