`tf.estimator.RunConfig` vs`tf.contrib.learn.RunConfig`

时间:2017-10-27 13:29:18

标签: tensorflow tensorflow-gpu

我很困惑我是否应该使用tf.estimator.RunConfigtf.contrib.learn.RunConfigRunConfig传递给估算工具。

使用tf.contrib.learn.RunConfig非常简单:

rc = tf.contrib.learn.RunConfig(save_checkpoints_secs=1,
                                model_dir=model_dir)

但是tf.estimator.RunConfig有一些奇怪的语法:

rc = tf.estimator.RunConfig()
rc = rc.replace(save_checkpoints_secs=1,
                model_dir=model_dir)

是否有理由偏好一个RunConfig而不是另一个?文档不清楚。

1 个答案:

答案 0 :(得分:0)

请选择核心版本,即tf.estimator.RunConfigtf.contrib.learn.*很快就会被弃用。 顺便说一句,你可以tf.estimator.RunConfig(save_check...)