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
HELSON c6a1a62636
[hotfix] fix zero's incompatibility with checkpoint in torch-1.12 (#1786)
2 years ago
..
__init__.py [zero] add chunk init function for users (#1729) 2 years ago
chunk.py [hotfix] fix zero's incompatibility with checkpoint in torch-1.12 (#1786) 2 years ago
manager.py [hotfix] fix zero's incompatibility with checkpoint in torch-1.12 (#1786) 2 years ago
search_utils.py [zero] add chunk init function for users (#1729) 2 years ago
utils.py [zero] add chunk init function for users (#1729) 2 years ago