From e0593efa4ff95ac50a11833f3cda902c6ecfecf7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Lucas=20de=20Sousa=20Almeida?= Date: Fri, 7 Mar 2025 09:49:48 -0300 Subject: [PATCH] Avoiding to copy a config file already placed in the logging dir MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: João Lucas de Sousa Almeida --- terratorch/cli_tools.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/terratorch/cli_tools.py b/terratorch/cli_tools.py index 2308e2ec..922e4ceb 100644 --- a/terratorch/cli_tools.py +++ b/terratorch/cli_tools.py @@ -342,8 +342,14 @@ def setup(self, trainer: Trainer, pl_module: LightningModule, stage: str) -> Non # broadcast so that all ranks are in sync on future calls to .setup() self.already_saved = trainer.strategy.broadcast(self.already_saved) + # Copying config file to log dir - shutil.copyfile(self.config_path_original, self.config_path_new) + # This is used to copy the exact original yaml file in the log + # directory in order to facilitate the reproducibility + if not os.path.samefile(self.config_path_original, self.config_path_new): + # When the file being used is already in the log directory, this + # operation is not necessary + shutil.copyfile(self.config_path_original, self.config_path_new) class StateDictAwareModelCheckpoint(ModelCheckpoint): # necessary as we wish to have one model checkpoint with only state dict and one with standard lightning checkpoints