.. |
__init__.py
|
98f5131bb6
add test model Megatron_GPT2
|
4 years ago |
ds_config_func_bs4_zero1.json
|
41db1c2f03
ZeRO-Offload release (#391)
|
4 years ago |
ds_config_func_bs4_zero2.json
|
41db1c2f03
ZeRO-Offload release (#391)
|
4 years ago |
ds_config_func_bs4_zero2_offload.json
|
41db1c2f03
ZeRO-Offload release (#391)
|
4 years ago |
ds_config_func_bs8_no_zero.json
|
41db1c2f03
ZeRO-Offload release (#391)
|
4 years ago |
ds_config_func_bs8_zero0_gas3.json
|
41db1c2f03
ZeRO-Offload release (#391)
|
4 years ago |
ds_config_func_bs8_zero1.json
|
41db1c2f03
ZeRO-Offload release (#391)
|
4 years ago |
ds_config_func_bs8_zero2.json
|
41db1c2f03
ZeRO-Offload release (#391)
|
4 years ago |
ds_config_func_bs8_zero2_gas3.json
|
41db1c2f03
ZeRO-Offload release (#391)
|
4 years ago |
ds_config_func_bs8_zero2_offload.json
|
41db1c2f03
ZeRO-Offload release (#391)
|
4 years ago |
ds_config_func_scheduler.json
|
41db1c2f03
ZeRO-Offload release (#391)
|
4 years ago |
ds_config_perf_bs16.json
|
41db1c2f03
ZeRO-Offload release (#391)
|
4 years ago |
ds_config_perf_bs32.json
|
41db1c2f03
ZeRO-Offload release (#391)
|
4 years ago |
ds_config_perf_bs8.json
|
41db1c2f03
ZeRO-Offload release (#391)
|
4 years ago |
ds_gpt2_test.sh
|
41db1c2f03
ZeRO-Offload release (#391)
|
4 years ago |
run_checkpoint_test.py
|
41db1c2f03
ZeRO-Offload release (#391)
|
4 years ago |
run_func_test.py
|
41db1c2f03
ZeRO-Offload release (#391)
|
4 years ago |
run_perf_baseline.py
|
98f5131bb6
add test model Megatron_GPT2
|
4 years ago |
run_perf_test.py
|
98f5131bb6
add test model Megatron_GPT2
|
4 years ago |
test_common.py
|
7240abf355
Samyamr/grad acc stage2 (#338)
|
4 years ago |