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-17 16:08:30 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
e8acf55e8bd53b88f48e183ef360c8691a027e9a
ColossalAI/colossalai/utils
History
Jiarui Fang 9e4c6449b0 [checkpoint] add ColoOptimizer checkpointing (#1316)
2022-07-15 09:52:55 +08:00
..
checkpoint
[checkpoint] add ColoOptimizer checkpointing (#1316)
2022-07-15 09:52:55 +08:00
data_sampler
Refactored docstring to google style
2022-03-29 17:17:47 +08:00
model
[tensor] distributed checkpointing for parameters (#1240)
2022-07-12 15:51:06 +08:00
multi_tensor_apply
Refactored docstring to google style
2022-03-29 17:17:47 +08:00
profiler
[hotfix] remove potiential circle import (#1307)
2022-07-14 13:44:26 +08:00
tensor_detector
Refactored docstring to google style
2022-03-29 17:17:47 +08:00
__init__.py
[refactory] add nn.parallel module (#1068)
2022-06-06 15:34:41 +08:00
activation_checkpoint.py
[util] fixed activation checkpointing on torch 1.9 (#719)
2022-04-12 09:35:45 +08:00
checkpointing.py
polish checkpoint docstring (#637)
2022-04-02 13:34:33 +08:00
common.py
[hotfix] fix sharded optim step and clip_grad_norm (#1226)
2022-07-08 13:34:48 +08:00
cuda.py
[refactor] refactor the memory utils (#715)
2022-04-11 16:47:57 +08:00
memory.py
[gemini] APIs to set cpu memory capacity (#809)
2022-04-19 16:05:22 +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
Powered by Gitea Version: 1.25.2 Page: 3685ms Template: 455ms
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