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
0442f940f021d024ca390485f0cdf0856fe6cb36
ColossalAI/colossalai/nn
History
ver217 04c9a86af8 [zero] ZeroDDP supports controlling outputs' dtype (#1399)
2022-08-02 17:49:11 +08:00
..
_ops
[colotensor] add Tensor.view op and its unit test (#1343)
2022-07-21 10:53:15 +08:00
graph
[refactor] move process group from _DistSpec to ColoTensor. (#1203)
2022-07-06 16:15:16 +08:00
layer
[NFC] polish colossalai/nn/layer/wrapper/pipeline_wrapper.py code style (#1303)
2022-07-13 19:01:07 +08:00
loss
[tensor] add unitest for colo_tensor 1DTP cross_entropy (#1230)
2022-07-07 19:17:23 +08:00
lr_scheduler
[NFC] polish colossalai/nn/lr_scheduler/onecycle.py code style (#1269)
2022-07-13 12:08:21 +08:00
metric
[hotfix] Raise messages for indivisible batch sizes with tensor parallelism (#622)
2022-04-02 16:12:04 +08:00
optimizer
[hotfix] adapt ProcessGroup and Optimizer to ColoTensor (#1388)
2022-07-29 19:33:24 +08:00
parallel
[zero] ZeroDDP supports controlling outputs' dtype (#1399)
2022-08-02 17:49:11 +08:00
__init__.py
[pipeline] refactor the pipeline module (#1087)
2022-06-10 11:27:38 +08:00
init.py
[NFC] polish colossalai/nn/init.py code style (#1292)
2022-07-13 10:51:55 +08:00
Powered by Gitea Version: 1.25.2 Page: 1877ms Template: 196ms
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