fix logger in callback

mindspore-lab · Jun 28, 2023 · 3c08a3c · 3c08a3c
1 parent 2b95dc0
commit 3c08a3c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/mindocr/utils/callbacks.py b/mindocr/utils/callbacks.py
@@ -66,7 +66,12 @@ def __init__(
         self.loader_eval = loader
         self.network = network
         self.ema = ema
-        self.logger = Logger("mindocr")
+
+        if logger is None:
+            self.logger = Logger("mindocr", rank=self.rank_id)
+        else:
+            self.logger = logger
+
         self.val_interval = val_interval
         self.val_start_epoch = val_start_epoch
         self.log_interval = log_interval