| .. |
|
amp
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
builder
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +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
|
fix pipeline forward return tensors (#176)
|
2022-01-21 15:46:02 +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
|
Develop/experiments (#59)
|
2021-12-09 15:08:29 +08:00 |
|
trainer
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
utils
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +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 docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
708404d5f8