Skip to content

Commit

Permalink
added reduce ddp results on eval
Browse files Browse the repository at this point in the history
  • Loading branch information
williamFalcon committed Jun 30, 2020
1 parent aa6b42b commit 9f1ed61
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions pytorch_lightning/trainer/evaluation_loop.py
Original file line number Diff line number Diff line change
Expand Up @@ -353,6 +353,10 @@ def _evaluate(
return eval_results

def reduce_eval_ddp(self, eval_results):
# ignore bad inputs
if len(eval_results) or eval_results is None:
return

for k, v in eval_results.items():
if isinstance(v, dict):
self.reduce_eval_ddp(v)
Expand Down

0 comments on commit 9f1ed61

Please sign in to comment.