modify torch cpp log level to Error to avoid annoying print

ultralytics · May 8, 2024 · 04bfde8 · 04bfde8
1 parent 920c721
commit 04bfde8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/export.py b/export.py
@@ -48,6 +48,7 @@
 import contextlib
 import json
 import os
+os.environ["TORCH_CPP_LOG_LEVEL"] = "ERROR"
 import platform
 import re
 import subprocess