basilshaji
commited on
Update yolov5/utils/torch_utils.py
Browse files
yolov5/utils/torch_utils.py
CHANGED
@@ -17,7 +17,7 @@ import torch.nn as nn
|
|
17 |
import torch.nn.functional as F
|
18 |
from torch.nn.parallel import DistributedDataParallel as DDP
|
19 |
|
20 |
-
from utils.general import LOGGER, check_version, colorstr, file_date, git_describe
|
21 |
|
22 |
LOCAL_RANK = int(os.getenv("LOCAL_RANK", -1)) # https://pytorch.org/docs/stable/elastic/run.html
|
23 |
RANK = int(os.getenv("RANK", -1))
|
|
|
17 |
import torch.nn.functional as F
|
18 |
from torch.nn.parallel import DistributedDataParallel as DDP
|
19 |
|
20 |
+
from yolov5.utils.general import LOGGER, check_version, colorstr, file_date, git_describe
|
21 |
|
22 |
LOCAL_RANK = int(os.getenv("LOCAL_RANK", -1)) # https://pytorch.org/docs/stable/elastic/run.html
|
23 |
RANK = int(os.getenv("RANK", -1))
|