首頁  >  文章  >  後端開發  >  如何控制終端中的 TensorFlow 調試輸出?

如何控制終端中的 TensorFlow 調試輸出?

Patricia Arquette
Patricia Arquette原創
2024-10-26 09:58:30156瀏覽

How to Control TensorFlow Debugging Output in Your Terminal?

在終端機中控制TensorFlow 偵錯資訊

使用TensorFlow 時,終端中會顯示各種偵錯資訊,包括有關載入的程式庫和內容的詳細資訊檢測到的設備。雖然此資訊對於調試很有用,但它可能會變得令人不知所措或分散注意力。為了解決這個問題,TensorFlow 提供了一個自訂記錄偵錯資訊層級的機制。

停用偵錯訊息

要停用所有偵錯訊息,請將 TF_CPP_MIN_LOG_LEVEL 環境變數設為3. 這將抑制 TensorFlow 的所有資訊性訊息。

<code class="python">import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
import tensorflow as tf</code>

日誌等級自訂

TF_CPP_MIN_LOG_LEVEL 變數允許對日誌記錄等級進行更精細的控制。以下值代表不同的日誌記錄等級:

  • 0:記錄所有訊息(預設)。
  • 1:不列印 INFO 訊息。
  • 2:INFO和 WARNING 訊息不會被列印。
  • 3:不會列印 INFO、WARNING 和 ERROR 訊息。

示例

以下示例演示瞭如何抑制所有非錯誤消息:

<code class="python">os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
import tensorflow as tf</code>

測試版本

此解決方案已在TensorFlow 版本0.12 和1.0 上進行了測試。

以上是如何控制終端中的 TensorFlow 調試輸出?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn