You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/gemini/chunk
YH 1a229045af
Add interface for colo tesnor dp size (#3227)
2 years ago
..
__init__.py [Gemini] chunk init use OrderedParamGenerator (#2110) 2 years ago
chunk.py [zero] fix error for BEiT models (#2169) 2 years ago
manager.py Add interface for colo tesnor dp size (#3227) 2 years ago
search_utils.py [hotfix] fix chunk size can not be divided (#2867) 2 years ago
utils.py [gemini] update ddp strict mode (#2518) 2 years ago