Using the same name to save the config yaml file

Signed-off-by: João Lucas de Sousa Almeida <[email protected]>

Author: Joao-L-S-Almeida

terratorch/cli_tools.py

@@ -182,6 +182,8 @@ def setup(self, trainer: Trainer, pl_module: LightningModule, stage: str) -> Non
         if self.already_saved:
             return
 
+        _, self.config_filename = os.path.split(self.config.config[0].abs_path)
+
         if self.save_to_log_dir:
             log_dir = trainer.log_dir or trainer.default_root_dir  # this broadcasts the directory
             if log_dir is None: