From 7479070da262ad3153c53b54840d4094bec756da Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 3 Nov 2021 17:17:38 +0100 Subject: [PATCH] Update torch_utils.py import `LOGGER` (#5483) --- utils/torch_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/torch_utils.py b/utils/torch_utils.py index ed399a70a5c..6e619d9c695 100644 --- 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):