/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
colossalai/utils/__init__.py
30 строк
660 B
Hongxin Liu
[checkpointio] fix hybrid plugin model save (#6106)
31 окт 2024, 12:04
Не верифицирован
31 окт 2024, 12:04
c2e8f61
Код
Авторство
О чём код?
from .common import ( _cast_float, conditional_context, disposable, ensure_path_exists, free_storage, get_current_device, get_non_persistent_buffers_set, is_ddp_ignored, set_seed, ) from .multi_tensor_apply import multi_tensor_applier from .tensor_detector import TensorDetector from .timer import MultiTimer, Timer __all__ = [ "conditional_context", "Timer", "MultiTimer", "multi_tensor_applier", "TensorDetector", "ensure_path_exists", "disposable", "_cast_float", "free_storage", "set_seed", "get_current_device", "is_ddp_ignored", "get_non_persistent_buffers_set", ]