.. |
__init__.py
|
2e99f6edf6
[DRAFT] Tentative implementation of MiCS (#2964)
|
1 年之前 |
config.py
|
1e0c39c6bf
enable pipeline checkpoint loading mode (#3629)
|
1 年之前 |
contiguous_memory_allocator.py
|
389bf69319
fix: Remove duplicate word the (#4051)
|
1 年之前 |
linear.py
|
42858a9891
save tensors in context of memory_efficient_linear (#3413)
|
1 年之前 |
mics.py
|
1e0c39c6bf
enable pipeline checkpoint loading mode (#3629)
|
1 年之前 |
mics_utils.py
|
2e99f6edf6
[DRAFT] Tentative implementation of MiCS (#2964)
|
1 年之前 |
offload_config.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
parameter_offload.py
|
0f5406323c
[CPU] FusedAdam and CPU training support (#3991)
|
1 年之前 |
partition_parameters.py
|
7f90ef4bdd
Multiple zero stage 3 related fixes (#3886)
|
1 年之前 |
partitioned_param_coordinator.py
|
977254c148
Disable z3 tracing profiler (#4106)
|
1 年之前 |
partitioned_param_profiler.py
|
d18aa2c79c
ZeRO++ (#3784)
|
1 年之前 |
stage3.py
|
1e0c39c6bf
enable pipeline checkpoint loading mode (#3629)
|
1 年之前 |
stage_1_and_2.py
|
2ef7300963
Bug fix
|
1 年之前 |
test.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
tiling.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
utils.py
|
d3550dc88a
Adagrad support in ZeRO (#3401)
|
1 年之前 |