From c960fdbd841d19bf36aaced6c10b69d229e42f3d Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 22 Jul 2022 11:54:31 +0200
Subject: [PATCH] Update hubconf.py to reset LOGGER.level after load (#8674)

Resolves silent outputs after model load
---
 hubconf.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/hubconf.py b/hubconf.py
index 6bb9484a856d..8748279e027a 100644
--- a/hubconf.py
+++ b/hubconf.py
@@ -34,6 +34,7 @@ def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbo
     from utils.general import LOGGER, check_requirements, intersect_dicts, logging
     from utils.torch_utils import select_device
 
+    level = LOGGER.level
     if not verbose:
         LOGGER.setLevel(logging.WARNING)
     check_requirements(exclude=('tensorboard', 'thop', 'opencv-python'))
@@ -57,6 +58,7 @@ def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbo
                     model.names = ckpt['model'].names  # set class names attribute
         if autoshape:
             model = AutoShape(model)  # for file/URI/PIL/cv2/np inputs and NMS
+        LOGGER.setLevel(level)
         return model.to(device)
 
     except Exception as e: