Update run_func_test.py

Renaming config files to gas3

Update run_func_test.py
Renaming config files to gas3
f4726b75 · Samyam Rajbhandari · GitHub · 6122a74e · f4726b75
Unverified Commit f4726b75 authored Aug 31, 2020 by Samyam Rajbhandari Committed by GitHub Aug 31, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tests/model/Megatron_GPT2/run_func_test.py tests/model/Megatron_GPT2/run_func_test.py +2 -2

No files found.
--- a/tests/model/Megatron_GPT2/run_func_test.py
+++ b/tests/model/Megatron_GPT2/run_func_test.py
@@ -210,8 +210,8 @@ class GPT2FuncTestCase(BaseTestCase):
            "seq_length": SEQ_LEN,
            "heads": ATTN_HEADS,
            "deepspeed": True,
-            "json": "ds_config_func_bs8_zero2_gas10.json",
-            "baseline": "ds_config_func_bs8_zero0_gas10.json",
+            "json": "ds_config_func_bs8_zero2_gas3.json",
+            "baseline": "ds_config_func_bs8_zero0_gas3.json",
        }

        succ = self.run_test(test_config, 0.01)