henry000 commited on
Commit
fc29438
Β·
1 Parent(s): 593a50f

πŸ› [Fix] bug in DDP sync_dist

Browse files
yolo/tools/solver.py CHANGED
@@ -49,10 +49,7 @@ class ValidateModel(BaseModel):
49
  "map": batch_metrics["map"],
50
  "map_50": batch_metrics["map_50"],
51
  },
52
- prog_bar=True,
53
- logger=False,
54
  on_step=True,
55
- sync_dist=True,
56
  batch_size=batch_size,
57
  )
58
  return predicts
 
49
  "map": batch_metrics["map"],
50
  "map_50": batch_metrics["map_50"],
51
  },
 
 
52
  on_step=True,
 
53
  batch_size=batch_size,
54
  )
55
  return predicts
yolo/utils/bounding_box_utils.py CHANGED
@@ -129,7 +129,7 @@ def generate_anchors(image_size: List[int], strides: List[int]):
129
  shift = stride // 2
130
  h = torch.arange(0, H, stride) + shift
131
  w = torch.arange(0, W, stride) + shift
132
- if torch.__version__ >= "1.10.0":
133
  anchor_h, anchor_w = torch.meshgrid(h, w, indexing="ij")
134
  else:
135
  anchor_h, anchor_w = torch.meshgrid(h, w)
 
129
  shift = stride // 2
130
  h = torch.arange(0, H, stride) + shift
131
  w = torch.arange(0, W, stride) + shift
132
+ if torch.__version__ >= "2.3.0":
133
  anchor_h, anchor_w = torch.meshgrid(h, w, indexing="ij")
134
  else:
135
  anchor_h, anchor_w = torch.meshgrid(h, w)