From 28e06aa3911ecf8bcfb9f13aa3be378e4ab62593 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 12 May 2024 22:16:54 +0200 Subject: [PATCH] Centralize ENV variable definition in utils/general.py (#13004) * Centralize ENV variable definition in utils/general.py Signed-off-by: Glenn Jocher * Update export.py Signed-off-by: Glenn Jocher * Auto-format by https://ultralytics.com/actions --------- Signed-off-by: Glenn Jocher Co-authored-by: UltralyticsAssistant --- export.py | 2 -- utils/general.py | 2 ++ 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/export.py b/export.py index cd2202c9c072..214d903c2998 100644 --- 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 index 5a9325eec757..64ae50626e14 100644 --- 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=""):