From e4ce9af45c047cee3f44c216e36390b38b2c4919 Mon Sep 17 00:00:00 2001 From: lruizcalico Date: Thu, 20 Jun 2024 19:52:31 -0700 Subject: [PATCH] upload train valid folders --- src/baskerville/trainer.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/baskerville/trainer.py b/src/baskerville/trainer.py index 33ec730..16a89d7 100644 --- a/src/baskerville/trainer.py +++ b/src/baskerville/trainer.py @@ -510,8 +510,8 @@ def eval_step1_distr(xd, yd): # upload to gcs if self.gcs: - upload_folder_gcs(self.log_dir, self.gcs_log_dir) - + upload_folder_gcs(train_log_dir, self.gcs_log_dir + "/train") + upload_folder_gcs(valid_log_dir, self.gcs_log_dir + "/valid") # checkpoint managers[di].save() model.save( @@ -713,7 +713,8 @@ def eval_step_distr(xd, yd): # upload to gcs if self.gcs: - upload_folder_gcs(self.log_dir, self.gcs_log_dir) + upload_folder_gcs(train_log_dir, self.gcs_log_dir + "/train") + upload_folder_gcs(valid_log_dir, self.gcs_log_dir + "/valid") # checkpoint manager.save()