Skip to content

GitLab

Unverified Commit f4726b75 authored by Samyam Rajbhandari's avatar Samyam Rajbhandari Committed by GitHub
Browse files

Update run_func_test.py 

Renaming config files to gas3
parent 6122a74e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/model/Megatron_GPT2/run_func_test.py
View file @ f4726b75
...@@ -210,8 +210,8 @@ class GPT2FuncTestCase(BaseTestCase): ...@@ -210,8 +210,8 @@ class GPT2FuncTestCase(BaseTestCase):
"seq_length": SEQ_LEN, "seq_length": SEQ_LEN,
"heads": ATTN_HEADS, "heads": ATTN_HEADS,
"deepspeed": True, "deepspeed": True,
"json": "ds_config_func_bs8_zero2_gas10.json", "json": "ds_config_func_bs8_zero2_gas3.json",
"baseline": "ds_config_func_bs8_zero0_gas10.json", "baseline": "ds_config_func_bs8_zero0_gas3.json",
} }
succ = self.run_test(test_config, 0.01) succ = self.run_test(test_config, 0.01)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment