Update torch_utils.py import LOGGER (#5483)

ultralytics · Nov 3, 2021 · 84a8099 · 84a8099
1 parent df30426
commit 84a8099
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/torch_utils.py b/utils/torch_utils.py
@@ -20,13 +20,13 @@
 import torch.nn.functional as F
 import torchvision
 
+from utils.general import LOGGER
+
 try:
     import thop  # for FLOPs computation
 except ImportError:
     thop = None
 
-LOGGER = logging.getLogger(__name__)
-
 
 @contextmanager
 def torch_distributed_zero_first(local_rank: int):