github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-11 14:43:10 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
1653063fce7062dfd652e73691fac44a24698cd5
ColossalAI/colossalai/nn/layer/base_layer.py
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

2.8 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink