diff --git a/didactic/config/experiment/finetune.yaml b/didactic/config/experiment/finetune.yaml
index 58dbc8b..e3b12d6 100644
--- 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
index da579a4..9f3b492 100644
--- 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
index 2aaff31..6c0dd2f 100644
--- 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: