main.yml
Commit
3 years ago
Author
- Files21
- .github/workflows
DeepSpeedExamples- deepspeed
- runtime
- activation_checkpointing
checkpointing.py
- engine.py
- fp16
- fused_optimizer.py
- unfused_optimizer.py
- pipe
- engine.py
- module.py
- p2p.py
- state_dict_factory.py
- utils.py
- zero
- partition_parameters.py
- stage1.py
- stage2.py
- stage3.py
- utils
- zero_to_fp32.py
- docs
Gemfile.lock
- tests/unit
- megatron_model.py
- test_configurable_parallel.py
- test_fp16.py
- test_zero_context.py