github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-11-30 07:14:40 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
24586599193d4f18fbaf66174f6fd669d59bc9d2
ColossalAI/colossalai/nn/parallel/data_parallel.py
HELSON 2458659919 [zero] fix error for BEiT models (#2169) 
* [zero] fix error for BEiT models

* [ColoParameter] add unpack operation for tuple arguments

* fix bugs

* fix chunkv2 unit testing

* add assertion for gradient state
2022-12-26 15:03:54 +08:00

26 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink