.. |
activation_checkpointing
|
42c1e916f6
feat(activation_checkpointing): add `non_reentrant_checkpoint` to support inputs require no grad (#4118)
|
1 年之前 |
comm
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
half_precision
|
57a27b0803
add type checker ignore to resolve that pylance can't resolved noqa annotation (#4102)
|
1 年之前 |
pipe
|
60a3e89eed
use ```non_reentrant_checkpoint``` fix requires_grad of input must be true for activation checkpoint layer in pipeline train. (#4224)
|
1 年之前 |
sparse_tensor
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
utils
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
zero
|
9adc73ff65
Handle empty parameter groups (#4277)
|
1 年之前 |
test_autocast.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
test_data.py
|
aef6c65ce3
Reduce Unit Test Times (Part 3) (#3850)
|
1 年之前 |
test_data_efficiency.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
test_ds_config_dict.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
test_ds_config_model.py
|
604d701e35
Introduce pydantic_v1 compatibility module for pydantic>=2.0.0 support (#4407)
|
1 年之前 |
test_ds_initialize.py
|
6b2365e4fa
Re-enable elastic training for torch 2+ (#4010)
|
1 年之前 |
test_lr_schedulers.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
test_multi_output_model.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
test_mup_optimizers.py
|
9647ea791d
Add MuP optimizers (#2043)
|
1 年之前 |
test_pld.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
test_runtime_utils.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |