當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python tf.estimator.LoggingTensorHook用法及代碼示例


每 N 個局部步驟、每 N 秒或最後打印給定的張量。

繼承自:SessionRunHook

用法

tf.estimator.LoggingTensorHook(
    tensors, every_n_iter=None, every_n_secs=None, at_end=False, formatter=None
)

參數

  • tensors dict 將 string-valued 標簽映射到張量/張量名稱,或 iterable 的張量/張量名稱。
  • every_n_iter int ,在當前工作人員上每執行 N 個本地步驟,打印一次 tensors 的值。
  • every_n_secs intfloat ,每 N 秒打印一次 tensors 的值。應該提供every_n_iterevery_n_secs 中的一個。
  • at_end bool 指定是否在運行結束時打印 tensors 的值。
  • formatter 函數,接受 tag -> Tensor 的 dict 並返回一個字符串。如果 None 使用默認打印所有張量。

拋出

  • ValueError 如果every_n_iter 是非正數。

遷移到 TF2

警告:這個 API 是為 TensorFlow v1 設計的。繼續閱讀有關如何從該 API 遷移到本機 TensorFlow v2 等效項的詳細信息。見TensorFlow v1 到 TensorFlow v2 遷移指南有關如何遷移其餘代碼的說明。

請查看此筆記本以了解如何將 API 遷移到 TF2。

張量將以INFO 嚴重性打印到日誌中。如果您沒有看到日誌,您可能需要在導入後添加以下行:

tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.INFO)

請注意,如果 at_end 為 True,則 tensors 不應包含其評估會產生副作用(例如消耗額外輸入)的任何張量。

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.estimator.LoggingTensorHook。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。