remove batch training script

talmolab · Sep 27, 2024 · 6711697 · 6711697
1 parent c721e90
commit 6711697
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 61 deletions.
diff --git a/dreem/training/configs/base.yaml b/dreem/training/configs/base.yaml
@@ -114,7 +114,7 @@ logging:
   group: "example"
   save_dir: './logs'
   project: "GTR"
-  log_model: "all"
+  log_model: null
 
 early_stopping:
   monitor: "val_loss"
@@ -131,12 +131,12 @@ checkpointing:
   save_last: true
   dirpath: null
   auto_insert_metric_name: true
-  every_n_epochs: 10
+  every_n_epochs: 1
 
 trainer:
-  # only use mps and devices params for apple silicon runs
-  accelerator: "mps"
-  devices: 1
+  # only use this for local apple silicon runs; change for cluster runs
+  # accelerator: "mps"
+  # devices: 1
   check_val_every_n_epoch: 1
   enable_checkpointing: true
   gradient_clip_val: null

diff --git a/dreem/training/configs/run_batch_job.py b/dreem/training/configs/run_batch_job.py