我想一些TensorBoard記錄添加到使用新的API tf.estimator的典範。Tensorflow - 使用tf.summary 1.2估算API
我有一個鉤設置像這樣:
summary_hook = tf.train.SummarySaverHook(
save_secs=2,
output_dir=MODEL_DIR,
summary_op=tf.summary.merge_all())
# ...
classifier.train(
input_fn,
steps=1000,
hooks=[summary_hook])
在我model_fn
,我也創建一個summary
-
然而,當我運行此代碼,我得到以下錯誤來自summary_hook
: Exactly one of scaffold or summary_op must be provided.
這可能是因爲tf.summary.merge_all()
未找到任何摘要並且正在返回None
,儘管有tf.summary.scalar
我在model_fn
中宣佈。
任何想法,爲什麼這將無法正常工作?
我有幾乎相同的代碼結構,但我得到了以下錯誤:「類型錯誤:獲取參數具有無效類型,必須是字符串或張量。 (不能將腳手架轉換爲張量或操作。)「任何想法爲什麼它不工作?謝謝! –
EXP0
在Tensorflow的新版本中,如果您將鉤子傳遞給」評估「,但不適用於」火車「,則此工作正常。對於列車只需要在'mode_fn'定義總結,他們將被自動記錄到tensorboard。(是的,我也覺得這是愚蠢的和違反直覺) – GPhilo