Skip to content

Commit 5ade91a

Browse files
committed
ADLR/megatron-lm!2136 - tests: Fix GPT test
1 parent 08e80b0 commit 5ade91a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/functional_tests/test_cases/gpt/gpt3_15b_8t_release/model_config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ MODEL_ARGS:
2525
--micro-batch-size: 4
2626
--rampup-batch-size: "384 384 97656250"
2727
--global-batch-size: 1152
28-
--train-samples: 4882812
28+
--train-samples: 19531250
2929
--manual-gc: true
3030

3131
# Transformer Engine args

0 commit comments

Comments
 (0)