/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
colossalai/zero/__init__.py
13 строк
390 B
Wenhao Chen
[shardformer]: support gpt-j, falcon, Mistral and add interleaved pipeline for bert (#5088)
28 ноя 2023, 11:54
Не верифицирован
28 ноя 2023, 11:54
7172459
Код
Авторство
О чём код?
from .gemini import GeminiAdamOptimizer, GeminiDDP, GeminiOptimizer, get_static_torch_model from .low_level import LowLevelZeroOptimizer from .wrapper import zero_model_wrapper, zero_optim_wrapper __all__ = [ "GeminiDDP", "GeminiOptimizer", "GeminiAdamOptimizer", "zero_model_wrapper", "zero_optim_wrapper", "LowLevelZeroOptimizer", "get_static_torch_model", ]