github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-10-11 18:01:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
1aaeb596c63752071cbbaa2477a7f2406901b70b
ColossalAI/colossalai/gemini/chunk
History
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
..
__init__.py
[Gemini] chunk init use OrderedParamGenerator (#2110)
2022-12-11 21:41:13 +08:00
chunk.py
[zero] fix error for BEiT models (#2169)
2022-12-26 15:03:54 +08:00
manager.py
[NFC] update chunk manager API (#2119)
2022-12-12 16:57:22 +08:00
search_utils.py
[Gemini] chunk init using runtime visited param order (#2115)
2022-12-12 18:06:16 +08:00
utils.py
[Gemini] chunk init using runtime visited param order (#2115)
2022-12-12 18:06:16 +08:00