DeepSpeed
53ac7947 - reduce size of megatron tests (#223)

Commit

5 years ago

reduce size of megatron tests (#223)

References

#223 - reduce size of model tests

Author

jeffra

jeffra

Parents

Files6

tests/model
- Megatron_GPT2
  - ds_config_func_bs4_zero2.json
  - ds_config_func_bs8_zero2.json
  - ds_gpt2_test.sh
  - run_checkpoint_test.py
  - run_func_test.py
- run_sanity_check.py

Loading