π [Fix] mul-GPU @ valid bugs, return mAP @ rk 0
Browse files- 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()
|