From 04bfde8942b5e3469c57387191ee2a79064874ed Mon Sep 17 00:00:00 2001 From: inisis Date: Wed, 8 May 2024 11:12:28 +0800 Subject: [PATCH 1/2] modify torch cpp log level to Error to avoid annoying print --- export.py | 1 + 1 file changed, 1 insertion(+) diff --git a/export.py b/export.py index 214d903c2998..d2ddba86c638 100644 --- 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 From 00d0affb15c982ad2cd2a73a49f5b93e3500b5ed Mon Sep 17 00:00:00 2001 From: UltralyticsAssistant Date: Wed, 8 May 2024 03:15:25 +0000 Subject: [PATCH 2/2] Auto-format by https://ultralytics.com/actions --- export.py | 1 + 1 file changed, 1 insertion(+) diff --git a/export.py b/export.py index d2ddba86c638..cd2202c9c072 100644 --- 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