diff --git a/scripts/stage3_train.sh b/scripts/stage3_train.sh
index 8fd5662..7275ee8 100644
--- a/scripts/stage3_train.sh
+++ b/scripts/stage3_train.sh
@@ -1 +1 @@
-python stage3.py --filename stage3 --mode pretrain --stage2_path all_checkpoints/stage2/epoch=09.ckpt --max_steps 400000
+python stage3.py --filename stage3 --mode train --stage2_path all_checkpoints/stage2/epoch=09.ckpt --max_steps 400000