Commit
•
84a8099
1
Parent(s):
df30426
Update torch_utils.py import `LOGGER` (#5483)
Browse files- utils/torch_utils.py +2 -2
utils/torch_utils.py
CHANGED
|
@@ -20,13 +20,13 @@ import torch.nn as nn
|
|
| 20 |
import torch.nn.functional as F
|
| 21 |
import torchvision
|
| 22 |
|
|
|
|
|
|
|
| 23 |
try:
|
| 24 |
import thop # for FLOPs computation
|
| 25 |
except ImportError:
|
| 26 |
thop = None
|
| 27 |
|
| 28 |
-
LOGGER = logging.getLogger(__name__)
|
| 29 |
-
|
| 30 |
|
| 31 |
@contextmanager
|
| 32 |
def torch_distributed_zero_first(local_rank: int):
|
|
|
|
| 20 |
import torch.nn.functional as F
|
| 21 |
import torchvision
|
| 22 |
|
| 23 |
+
from utils.general import LOGGER
|
| 24 |
+
|
| 25 |
try:
|
| 26 |
import thop # for FLOPs computation
|
| 27 |
except ImportError:
|
| 28 |
thop = None
|
| 29 |
|
|
|
|
|
|
|
| 30 |
|
| 31 |
@contextmanager
|
| 32 |
def torch_distributed_zero_first(local_rank: int):
|