代码之家  ›  专栏  ›  技术社区  ›  bluesummers

禁用tf上的TensorBoard记录。估计器方法

  •  0
  • bluesummers  · 技术社区  · 6 年前

    使用时,是否有办法禁用自动张力板记录 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 回复  |  直到 6 年前
        1
  •  2
  •   Arsey    5 年前

    使用 save_summary_steps=None

    推荐文章