| .. |
|
amp
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
builder
|
add pytorch hooks (#179)
|
2022-01-25 22:20:54 +08:00 |
|
communication
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
context
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
engine
|
add pytorch hooks (#179)
|
2022-01-25 22:20:54 +08:00 |
|
kernel
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
logging
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
nn
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
registry
|
add pytorch hooks (#179)
|
2022-01-25 22:20:54 +08:00 |
|
trainer
|
add pytorch hooks (#179)
|
2022-01-25 22:20:54 +08:00 |
|
utils
|
fixed utils docstring and add example to readme (#200)
|
2022-02-03 11:37:17 +08:00 |
|
zero
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
__init__.py
|
Develop/experiments (#59)
|
2021-12-09 15:08:29 +08:00 |
|
constants.py
|
Added MoE parallel (#127)
|
2022-01-07 15:08:36 +08:00 |
|
core.py
|
Develop/experiments (#59)
|
2021-12-09 15:08:29 +08:00 |
|
global_variables.py
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
initialize.py
|
fixed ddp bug on torch 1.8 (#194)
|
2022-01-28 15:14:04 +08:00 |
812357d63c