Browse Source

Merge pull request #778 from will-jl944/develop_jf

log only if local_rank is 0
FlyingQianMM 4 years ago
parent
commit
131dfea569
1 changed files with 2 additions and 2 deletions
  1. 2 2
      dygraph/paddlex/cv/models/base.py

+ 2 - 2
dygraph/paddlex/cv/models/base.py

@@ -312,10 +312,10 @@ class BaseModel:
                         eval_dataset,
                         batch_size=eval_batch_size,
                         return_details=False)
-                    logging.info('[EVAL] Finished, Epoch={}, {} .'.format(
-                        i + 1, dict2str(self.eval_metrics)))
                     # 保存最优模型
                     if local_rank == 0:
+                        logging.info('[EVAL] Finished, Epoch={}, {} .'.format(
+                            i + 1, dict2str(self.eval_metrics)))
                         best_accuracy_key = list(self.eval_metrics.keys())[0]
                         current_accuracy = self.eval_metrics[best_accuracy_key]
                         if current_accuracy > best_accuracy: