diff --git a/Makefile b/Makefile
index c0f51c7..6fddb7a 100644
--- a/Makefile
+++ b/Makefile
@@ -8,7 +8,7 @@ data: training.csv
 
 train: train.py training.csv
 	@echo "Training coordinate regressor..."
-	@bash -c 'source .venv/bin/activate && python train.py'
+	@bash -c 'source .venv/bin/activate && python train.py --epochs 50'
 
 eval: eval.py training.csv
 	@echo "Evaluating coordinate regressor..."
@@ -24,7 +24,7 @@ eval_frozen_encoder: eval.py training.csv
 
 train_frozen_layers: train.py training.csv
 	@echo "Training coordinate regressor with first transformer layers frozen..."
-	@bash -c 'source .venv/bin/activate && python train.py --output-path output_frozen_layers --freeze-transformer-layers 4'
+	@bash -c 'source .venv/bin/activate && python train.py --output-path output_frozen_layers --freeze-transformer-layers 4 --epochs 50'
 
 eval_frozen_layers: eval.py training.csv
 	@echo "Evaluating frozen-layer coordinate regressor..."