fix_resume_logger

Signed-off-by: Peng Zhang <pengz@uber.com>
horovod · Jan 20, 2022 · 0fe0b05 · 0fe0b05
1 parent a5edcd0
commit 0fe0b05
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/horovod/spark/lightning/remote.py b/horovod/spark/lightning/remote.py
@@ -123,8 +123,8 @@ def train(serialized_model):
                 train_logger = TensorBoardLogger(logs_path)
                 print(f"Setup logger: Using TensorBoardLogger: {train_logger}")
 
-            elif isinstance(logger, CometLogger) and logger._experiment_key is None:
-                # Resume logger experiment key if passed correctly from CPU.
+            elif isinstance(logger, CometLogger) and logger_experiment_key is None:
+                # Resume logger experiment with new log path if key passed correctly from CPU.
                 train_logger = CometLogger(
                     save_dir=logs_path,
                     api_key=logger.api_key,