Define training w.r.t. steps instead of epochs to follow XTab

creatis-myriad · Jan 16, 2025 · 16e87fa · 16e87fa
1 parent 282e4bc
commit 16e87fa
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/didactic/config/experiment/finetune.yaml b/didactic/config/experiment/finetune.yaml
@@ -7,7 +7,7 @@ defaults:
   - override /task/model/encoder: cardinal-ft-transformer
 
 trainer:
-  max_epochs: 500
+  max_steps: 500
 
 task:
   predict_losses:

diff --git a/didactic/config/experiment/pretrain.yaml b/didactic/config/experiment/pretrain.yaml
@@ -5,7 +5,7 @@ defaults:
   - override /task/model/encoder: cardinal-ft-transformer
 
 trainer:
-  max_epochs: 2500
+  max_steps: 2500
 
 task:
   contrastive_loss:

diff --git a/didactic/config/experiment/random.yaml b/didactic/config/experiment/random.yaml
@@ -5,7 +5,7 @@ defaults:
   - override /task/model/encoder: cardinal-ft-transformer
 
 trainer:
-  max_epochs: 2500
+  max_steps: 2500
 
 task:
   predict_losses: