Update torch_utils.py import LOGGER (#5483)

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):