ColossalAI/colossalai/nn
ver217 ae71036cd2
[utils] refactor parallel layers checkpoint and bcast model on loading checkpoint (#1548)
* refactor parallel layer

* broadcast rank0 model after load ckpt
2022-09-06 20:18:35 +08:00
..
_ops [embedding] tablewise sharding polish (#1535) 2022-09-02 11:09:37 +08:00
graph [NFC] polish doc style for ColoTensor (#1457) 2022-08-16 09:21:05 +08:00
layer [utils] refactor parallel layers checkpoint and bcast model on loading checkpoint (#1548) 2022-09-06 20:18:35 +08:00
loss
lr_scheduler
metric
optimizer
parallel [embedding] polish parallel embedding tablewise (#1545) 2022-09-06 10:41:20 +08:00
__init__.py
init.py