ColossalAI/colossalai/builder
ver217 8f02a88db2
add interleaved pipeline, fix naive amp and update pipeline model initializer (#80)
2021-12-20 23:26:19 +08:00
..
__init__.py add interleaved pipeline, fix naive amp and update pipeline model initializer (#80) 2021-12-20 23:26:19 +08:00
builder.py update examples and sphnix docs for the new api (#63) 2021-12-13 22:07:01 +08:00
pipeline.py add interleaved pipeline, fix naive amp and update pipeline model initializer (#80) 2021-12-20 23:26:19 +08:00