ColossalAI/colossalai/gemini/chunk
HELSON 6e4ac08172
[hotfix] fix chunk size can not be divided (#2867)
* [hotfix] fix chunk size can not be divided

* [hotfix] use numpy for python3.8
2023-02-22 15:04:46 +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 [hotfix] fix chunk size can not be divided (#2867) 2023-02-22 15:04:46 +08:00
search_utils.py [hotfix] fix chunk size can not be divided (#2867) 2023-02-22 15:04:46 +08:00
utils.py [gemini] update ddp strict mode (#2518) 2023-01-28 14:35:25 +08:00