From e785e931e30a7a0f193a68365d9e0a2871ddb670 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 12 May 2024 22:01:43 +0200 Subject: [PATCH 1/3] Centralize ENV variable definition in utils/general.py Signed-off-by: Glenn Jocher --- utils/general.py | 2 ++ 1 file changed, 2 insertions(+) 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=""): From 767b90dec668fbdc7175e8178b6be9d14f585fb5 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 12 May 2024 22:02:18 +0200 Subject: [PATCH 2/3] Update export.py Signed-off-by: Glenn Jocher --- export.py | 1 - 1 file changed, 1 deletion(-) diff --git a/export.py b/export.py index cd2202c9c072..d5c9d8dd495f 100644 --- a/export.py +++ b/export.py @@ -49,7 +49,6 @@ import json import os -os.environ["TORCH_CPP_LOG_LEVEL"] = "ERROR" import platform import re import subprocess From db7b4aeb689eeedf4d9a0651db8f20bee06a8db8 Mon Sep 17 00:00:00 2001 From: UltralyticsAssistant Date: Sun, 12 May 2024 20:02:39 +0000 Subject: [PATCH 3/3] Auto-format by https://ultralytics.com/actions --- export.py | 1 - 1 file changed, 1 deletion(-) diff --git a/export.py b/export.py index d5c9d8dd495f..214d903c2998 100644 --- a/export.py +++ b/export.py @@ -48,7 +48,6 @@ import contextlib import json import os - import platform import re import subprocess