github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-02 08:37:36 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
b09adff724c2bbded1c71cc51a707f736a0e2899
ColossalAI/colossalai/nn/_ops/linear.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

7.0 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink