ColossalAI/colossalai/gemini/chunk
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
chunk.py [zero] fix error for BEiT models (#2169) 2022-12-26 15:03:54 +08:00
manager.py
search_utils.py [Gemini] chunk init using runtime visited param order (#2115) 2022-12-12 18:06:16 +08:00
utils.py