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
dbdc9a778389128e51601806d92c756657dad67c
ColossalAI/colossalai
History
HELSON dbdc9a7783 added Multiply Jitter and capacity factor eval for MOE (#434)
2022-03-16 16:47:44 +08:00
..
amp
fixed fp16 optimizer none grad bug (#432)
2022-03-16 14:35:46 +08:00
builder
add pytorch hooks (#179)
2022-01-25 22:20:54 +08:00
communication
fix format (#332)
2022-03-11 15:50:28 +08:00
context
fixed bug in activation checkpointing test (#387)
2022-03-11 15:50:28 +08:00
engine
use double buffer to handle grad
2022-03-16 14:24:09 +08:00
kernel
[formart] format fixed for kernel\cuda_native codes (#335)
2022-03-11 15:50:28 +08:00
logging
[log] better logging display with rich (#426)
2022-03-16 09:51:15 +08:00
nn
added Multiply Jitter and capacity factor eval for MOE (#434)
2022-03-16 16:47:44 +08:00
registry
add pytorch hooks (#179)
2022-01-25 22:20:54 +08:00
trainer
Added profiler communication operations
2022-03-11 15:50:28 +08:00
utils
fixed mem monitor device (#433)
2022-03-16 15:25:02 +08:00
zero
sync before creating empty grad
2022-03-16 14:24:09 +08:00
__init__.py
Develop/experiments (#59)
2021-12-09 15:08:29 +08:00
constants.py
fix format constants.py (#358)
2022-03-11 15:50:28 +08:00
core.py
Develop/experiments (#59)
2021-12-09 15:08:29 +08:00
global_variables.py
Optimized MoE layer and fixed some bugs;
2022-03-11 15:50:28 +08:00
initialize.py
[fp16] refactored fp16 optimizer (#392)
2022-03-15 10:05:38 +08:00
Powered by Gitea Version: 1.25.2 Page: 968ms Template: 39ms
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