Skip to content

Commit

Permalink
Revert "Update torch_utils.py import LOGGER (#5483)"
Browse files Browse the repository at this point in the history
This reverts commit 84a8099.
  • Loading branch information
glenn-jocher authored Nov 3, 2021
1 parent 06bf8ef commit ddafc1b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions utils/torch_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit ddafc1b

Please sign in to comment.