Отключить ведение журнала TensorBoard в методах tf.Estimator - PullRequest
0 голосов
/ 06 декабря 2018

Есть ли способ отключить автоматическое ведение журнала TensorBoard при использовании класса tf.estimator.Estimator?

# 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 и не смог найти решение.

1 Ответ

0 голосов
/ 29 января 2019

Использование RunConfig для Estimator с save_summary_steps=None должно исправить это

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...