Tensorflow Graphics 大量依賴 L2 正規化張量,以及預期輸入值在特定範圍內的三角函數。在最佳化期間,更新可能會使這些變數取值,導致這些函數傳回 Inf
或 NaN
值。為了簡化此類問題的偵錯,TensorFlow Graphics 提供偵錯旗標,將斷言注入圖形中,以檢查正確的範圍和傳回值的有效性。由於這可能會減慢計算速度,因此預設情況下偵錯旗標設定為 False
。
使用者可以設定 -tfg_debug
旗標,以在偵錯模式下執行程式碼。也可以透過程式設計方式設定旗標,方法是先匯入這兩個模組
from absl import flags
from tensorflow_graphics.util import tfg_flags
然後將以下程式碼行新增至程式碼。
flags.FLAGS[tfg_flags.TFG_ADD_ASSERTS_TO_GRAPH].value = True