From 2f6b91ee8c038a59ac51fe1151e3defd6cd0f907 Mon Sep 17 00:00:00 2001 From: inisis <46103969+inisis@users.noreply.github.com> Date: Sun, 12 May 2024 23:10:18 +0800 Subject: [PATCH] Set `TORCH_CPP_LOG_LEVEL=ERROR` for reduced verbosity (#12989) * modify torch cpp log level to Error to avoid annoying print * Auto-format by https://ultralytics.com/actions --------- Co-authored-by: UltralyticsAssistant --- export.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/export.py b/export.py index 214d903c2..cd2202c9c 100644 --- a/export.py +++ b/export.py @@ -48,6 +48,8 @@ import argparse import contextlib import json import os + +os.environ["TORCH_CPP_LOG_LEVEL"] = "ERROR" import platform import re import subprocess