Logo
Explore Help
Register Sign In
github/ColossalAI
1
0
Fork 0
You've already forked ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-15 14:47:16 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
37cb70feecdcf7571850f254def6d0d623dd5097
ColossalAI/colossalai/utils
History
Jiarui Fang 05e33b2578 [zero] fix grad offload (#528)
* [zero] fix grad offload

* polish code
2022-03-25 18:23:25 +08:00
..
data_sampler
fixed utils docstring and add example to readme (#200)
2022-02-03 11:37:17 +08:00
gradient_accumulation
Fixed docstring in colossalai (#171)
2022-01-21 10:44:30 +08:00
memory_tracer
[zero] refactor model data tracing (#522)
2022-03-25 18:03:32 +08:00
memory_utils
[zero] fix grad offload (#528)
2022-03-25 18:23:25 +08:00
multi_tensor_apply
Fixed docstring in colossalai (#171)
2022-01-21 10:44:30 +08:00
profiler
fixed error when no collective communication in CommProfiler
2022-03-14 17:21:00 +08:00
tensor_detector
Added tensor detector (#393)
2022-03-14 18:01:46 +08:00
__init__.py
[memory] add model data tensor moving api (#503)
2022-03-24 14:29:41 +08:00
activation_checkpoint.py
Added activation offload (#331)
2022-03-11 15:50:28 +08:00
checkpointing.py
fixed mkdir conflict and align yapf config with flake (#220)
2022-02-15 11:31:13 +08:00
common.py
[MOE] remove old MoE legacy (#493)
2022-03-22 17:37:16 +08:00
cuda.py
Fixed docstring in colossalai (#171)
2022-01-21 10:44:30 +08:00
moe.py
[polish] polish singleton and global context (#500)
2022-03-23 18:03:39 +08:00
timer.py
[profiler] primary memory tracer
2022-03-11 15:50:28 +08:00
Powered by Gitea Version: 1.25.2 Page: 649ms Template: 57ms
English
Bahasa Indonesia Deutsch English Español Français Gaeilge Italiano Latviešu Magyar nyelv Nederlands Polski Português de Portugal Português do Brasil Suomi Svenska Türkçe Čeština Ελληνικά Български Русский Українська فارسی മലയാളം 日本語 简体中文 繁體中文(台灣) 繁體中文(香港) 한국어
Licenses API