| .. |
|
data_sampler
|
Refactored docstring to google style
|
2022-03-29 17:17:47 +08:00 |
|
gradient_accumulation
|
Refactored docstring to google style
|
2022-03-29 17:17:47 +08:00 |
|
memory_tracer
|
[zero] refactor memstats collector (#706)
|
2022-04-11 10:46:08 +08:00 |
|
memory_utils
|
[refactor] memory utils (#577)
|
2022-04-01 09:22:33 +08:00 |
|
multi_tensor_apply
|
Refactored docstring to google style
|
2022-03-29 17:17:47 +08:00 |
|
profiler
|
chen pcie profiler `to_tensorboard`
|
2022-04-11 14:28:02 +08:00 |
|
tensor_detector
|
Refactored docstring to google style
|
2022-03-29 17:17:47 +08:00 |
|
__init__.py
|
[zero] refactor memstats collector (#706)
|
2022-04-11 10:46:08 +08:00 |
|
activation_checkpoint.py
|
[hotfix] fix bugs in testing (#659)
|
2022-04-02 21:58:47 +08:00 |
|
checkpointing.py
|
polish checkpoint docstring (#637)
|
2022-04-02 13:34:33 +08:00 |
|
common.py
|
[zero] refactor memstats collector (#706)
|
2022-04-11 10:46:08 +08:00 |
|
cuda.py
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
moe.py
|
Refactored docstring to google style
|
2022-03-29 17:17:47 +08:00 |
|
timer.py
|
Refactored docstring to google style
|
2022-03-29 17:17:47 +08:00 |
cd9492854b