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/zero/gemini/chunk
Baizhou Zhang d99b2c961a
[hotfix] fix grad accumulation plus clipping for gemini (#5002)
1 year ago
..
__init__.py [misc] update pre-commit and run all files (#4752) 1 year ago
chunk.py [hotfix] fix grad accumulation plus clipping for gemini (#5002) 1 year ago
manager.py [gemini] support gradient accumulation (#4869) 1 year ago
search_utils.py [misc] update pre-commit and run all files (#4752) 1 year ago
utils.py [misc] update pre-commit and run all files (#4752) 1 year ago