basilshaji commited on
Commit
650ad4b
·
verified ·
1 Parent(s): a20ee3a

Update yolov5/utils/torch_utils.py

Browse files
Files changed (1) hide show
  1. yolov5/utils/torch_utils.py +1 -1
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))