首頁  >  文章  >  後端開發  >  如何靜音 TensorFlow 的調試輸出?

如何靜音 TensorFlow 的調試輸出?

DDD
DDD原創
2024-10-28 23:51:30188瀏覽

How to Silence TensorFlow's Debugging Output?

抑制 Tensorflow 偵錯輸出

Tensorflow 在初始化階段列印有關載入的程式庫、找到的裝置和其他偵錯資料的大量資訊。雖然此輸出對於故障排除很有價值,但它也可能令人難以承受並分散其他日誌訊息的注意力。本文探討如何停用此偵錯資訊。

為了抑制偵錯日誌的顯示,有一個簡單而有效的解決方案。透過將環境變數 TF_CPP_MIN_LOG_LEVEL 設定為適當的值,可以控制日誌記錄的等級。

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

使用 Tensorflow 版本 0.12 和 1.0 進行測試,將環境變數設為 3 可以有效停用所有偵錯輸出,只留下嚴重錯誤和警告。

為了更好地控制日誌記錄行為,可以設定不同的等級:

  • 0:記錄所有訊息(預設)
  • 1 :抑制訊息訊息
  • 2:抑制訊息和警告訊息
  • 3:抑制訊息、警告和錯誤訊息

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

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