fixes test issues on ddp (#1017)

* updated checkpoint docs * updated checkpoint docs * updated checkpoint docs * updated checkpoint docs * updated checkpoint docs * updated checkpoint docs * updated checkpoint docs * updated checkpoint docs * updated checkpoint docs * updated checkpoint docs * updated checkpoint docs * updated checkpoint docs * updated checkpoint docs * updated checkpoint docs * updated checkpoint docs * updated checkpoint docs * updated checkpoint docs
2026-09-13 12:50:26 +08:00 · 2020-03-02 21:50:38 -05:00
parent 5458d05cd8
commit 6dae5698ef
6 changed files with 72 additions and 1 deletions
@@ -349,7 +349,11 @@ class TrainerEvaluationLoopMixin(ABC):

        # log results of test
        if test_mode:
-            model.print(prog_bar_metrics)
+            if self.proc_rank == 0:
+                print('-' * 100)
+                print('TEST RESULTS')
+                print(prog_bar_metrics)
+                print('-' * 100)

        # log metrics
        self.log_metrics(log_metrics, {})