Merge pull request #910 from AI-Hypercomputer:aireen/fix_eval_step

PiperOrigin-RevId: 678416266
AI-Hypercomputer · Sep 24, 2024 · f5f0e29 · f5f0e29
2 parents 9a0cfd4 + c98d990
commit f5f0e29
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/end_to_end/tpu/test_convergence_1b_params.sh b/end_to_end/tpu/test_convergence_1b_params.sh
@@ -16,7 +16,7 @@ echo "Running test_convergence_1b_params.sh"
 # default values, can be override from command line
 export LOSS_THRESHOLD=100.0 # Set to large value so test is guaranteed to pass.
 export STEPS=20400 # Run for 20B tokens for a 1B sized mode for "chinchilla" scaling https://arxiv.org/abs/2203.15556
-export EVAL_STEPS=500
+export EVAL_STEPS=160
 export DATASET_TYPE=tfds
 
 # Set environment variables