Centralize ENV variable definition in utils/general.py (#13004)

* Centralize ENV variable definition in utils/general.py Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update export.py Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Auto-format by https://ultralytics.com/actions --------- Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
ultralytics · May 12, 2024 · 28e06aa · 28e06aa
1 parent d1ea6b1
commit 28e06aa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/export.py b/export.py
@@ -48,8 +48,6 @@
 import contextlib
 import json
 import os
-
-os.environ["TORCH_CPP_LOG_LEVEL"] = "ERROR"
 import platform
 import re
 import subprocess

diff --git a/utils/general.py b/utils/general.py
@@ -68,6 +68,8 @@
 os.environ["NUMEXPR_MAX_THREADS"] = str(NUM_THREADS)  # NumExpr max threads
 os.environ["OMP_NUM_THREADS"] = "1" if platform.system() == "darwin" else str(NUM_THREADS)  # OpenMP (PyTorch and SciPy)
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"  # suppress verbose TF compiler warnings in Colab
+os.environ["TORCH_CPP_LOG_LEVEL"] = "ERROR"  # suppress "NNPACK.cpp could not initialize NNPACK" warnings
+os.environ["KINETO_LOG_LEVEL"] = "5"  # suppress verbose PyTorch profiler output when computing FLOPs
 
 
 def is_ascii(s=""):