【发布时间】:2019-05-08 00:45:58
【问题描述】:
有没有办法在使用tf.estimator.Estimator 类时禁用自动 TensorBoard 日志记录?
# Classifier
classifier = tf.estimator.Estimator(
model_fn=lambda features, labels, mode, params: model(features, labels, mode, params, word_embeddings),
params=params,
model_dir=str(MODEL_DIR),
tensorboard=Fasle) # <---- Something like that?
for _ in range(params['epochs']):
classifier.train(input_fn=lambda: input_fn(generator, example_generator._data['train'] ,batch_size=params['batch_size']))
classifier.evaluate(input_fn=lambda: input_fn(generator, example_generator._data['validation'], batch_size=params['batch_size']))
我阅读了tf.estimator.RunConfig 并找不到解决方案。
【问题讨论】:
标签: tensorflow tensorboard tensorflow-estimator