diff --git a/utils/torch_utils.py b/utils/torch_utils.py
index ed399a7..6e619d9 100644
--- a/utils/torch_utils.py
+++ b/utils/torch_utils.py
@@ -20,13 +20,13 @@ import torch.nn as nn
 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):