henry000 commited on
Commit
a44a5c1
Β·
1 Parent(s): aee6ac8

πŸ› [Fix] mul-GPU @ valid bugs, return mAP @ rk 0

Browse files
Files changed (1) hide show
  1. yolo/tools/solver.py +2 -0
yolo/tools/solver.py CHANGED
@@ -256,6 +256,8 @@ class ModelValidator:
256
 
257
  with open(self.json_path, "w") as f:
258
  predict_json = collect_prediction(predict_json, self.progress.local_rank)
 
 
259
  json.dump(predict_json, f)
260
  if hasattr(self, "coco_gt"):
261
  self.progress.start_pycocotools()
 
256
 
257
  with open(self.json_path, "w") as f:
258
  predict_json = collect_prediction(predict_json, self.progress.local_rank)
259
+ if self.progress.local_rank != 0:
260
+ return
261
  json.dump(predict_json, f)
262
  if hasattr(self, "coco_gt"):
263
  self.progress.start_pycocotools()