You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/utils/model
YuliangLiu0306 c6930d8ddf
[pipelinable]use ColoTensor to replace dummy tensor. (#853)
3 years ago
..
colo_init_context.py [Tensor] init a tp network training unittest (#849) 3 years ago
pipelinable.py [pipelinable]use ColoTensor to replace dummy tensor. (#853) 3 years ago
utils.py [pipelinable]use pipelinable context to initialize non-pipeline model (#816) 3 years ago