[Fix]: Fix pytorch-lightning compatibility. (#309)

RangiLyu · Aug 22, 2021 · bd1fa26 · bd1fa26
1 parent f3ba459
commit bd1fa26
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/nanodet/trainer/task.py b/nanodet/trainer/task.py
@@ -217,7 +217,11 @@ def test_epoch_end(self, test_step_outputs):
         results = {}
         for res in test_step_outputs:
             results.update(res)
-        all_results = gather_results(results) if self.trainer.use_ddp else results
+        all_results = (
+            gather_results(results)
+            if dist.is_available() and dist.is_initialized()
+            else results
+        )
         if all_results:
             res_json = self.evaluator.results2json(all_results)
             json_path = os.path.join(self.cfg.save_dir, "results.json")