fix folder

calico · Jun 21, 2024 · 4acf8a2 · 4acf8a2
1 parent e4ce9af
commit 4acf8a2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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(train_log_dir, self.gcs_log_dir + "/train")
-                        upload_folder_gcs(valid_log_dir, self.gcs_log_dir + "/valid")
+                        upload_folder_gcs(train_log_dir, self.gcs_log_dir)
+                        upload_folder_gcs(valid_log_dir, self.gcs_log_dir)
                     # checkpoint
                     managers[di].save()
                     model.save(
@@ -713,8 +713,8 @@ def eval_step_distr(xd, yd):
 
                 # upload to gcs
                 if self.gcs:
-                    upload_folder_gcs(train_log_dir, self.gcs_log_dir + "/train")
-                    upload_folder_gcs(valid_log_dir, self.gcs_log_dir + "/valid")
+                    upload_folder_gcs(train_log_dir, self.gcs_log_dir)
+                    upload_folder_gcs(valid_log_dir, self.gcs_log_dir)
 
                 # checkpoint
                 manager.save()