cleanup

NRCan · Oct 11, 2024 · 7f2c5b0 · 7f2c5b0
1 parent d83f213
commit 7f2c5b0
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/geo_deep_learning/tools/mlflow_logger.py b/geo_deep_learning/tools/mlflow_logger.py
@@ -6,11 +6,7 @@
 class LoggerSaveConfigCallback(SaveConfigCallback):
     def save_config(self, trainer: Trainer, pl_module: LightningModule, stage: str) -> None:
         if isinstance(trainer.logger, MLFlowLogger):
-            print("Logging Config file to MLFlow.")
-            # config = self.parser.dump(self.config, skip_none=False)  # Required for proper reproducibility
-            # print(self.config.config[0])
             config_filepath = self.config.config[0]
-            print(type(config_filepath))
             trainer.logger.experiment.log_artifact(local_path=config_filepath, 
                                                    artifact_path="config",
                                                    run_id=trainer.logger.run_id)