我很困惑我是否应该使用tf.estimator.RunConfig
或tf.contrib.learn.RunConfig
将RunConfig
传递给估算工具。
使用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
而不是另一个?文档不清楚。
答案 0 :(得分:0)
请选择核心版本,即tf.estimator.RunConfig
。
tf.contrib.learn.*
很快就会被弃用。
顺便说一句,你可以tf.estimator.RunConfig(save_check...)