ColossalAI/colossalai/utils/model at 62f059251bfcfffcc2f1d89303e27628e5ea8d04 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-08 14:46:09 +00:00

Files

History

Jiarui Fang 62f059251b [Tensor] init a tp network training unittest (#849 )

2022-04-24 16:43:44 +08:00

..

colo_init_context.py

[Tensor] init a tp network training unittest (#849 )

2022-04-24 16:43:44 +08:00

pipelinable.py

[pipelinable]use pipelinable context to initialize non-pipeline model (#816 )

2022-04-24 13:03:12 +08:00

utils.py

[pipelinable]use pipelinable context to initialize non-pipeline model (#816 )

2022-04-24 13:03:12 +08:00