How can I use tensorboard with tf.estimator.Estimator

前端 未结 4 990
粉色の甜心
粉色の甜心 2020-12-30 06:44

I am considering to move my code base to tf.estimator.Estimator, but I cannot find an example on how to use it in combination with tensorboard summaries.

MWE:

<
4条回答
  •  轻奢々
    轻奢々 (楼主)
    2020-12-30 07:40

    For me this worked without adding any hooks or merge_all calls. I just added some tf.summary.image(...) in my model_fn and when I train the model they magically appear in tensorboard. Not sure what the exact mechanism is, however. I'm using TensorFlow 1.4.

提交回复
热议问题