/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
colossalai/testing/__init__.py
39 строк
865 B
Hongxin Liu
[test] merge old components to test to model zoo (#4945)
20 окт 2023, 05:35
Не верифицирован
20 окт 2023, 05:35
b8e770c
Код
Авторство
О чём код?
from .comparison import ( assert_close, assert_close_loose, assert_equal, assert_equal_in_group, assert_hf_output_close, assert_not_equal, check_state_dict_equal, ) from .pytest_wrapper import run_on_environment_flag from .utils import ( DummyDataloader, clear_cache_before_run, free_port, parameterize, rerun_if_address_is_in_use, rerun_on_exception, skip_if_not_enough_gpus, spawn, ) __all__ = [ "assert_equal", "assert_not_equal", "assert_close", "assert_close_loose", "assert_equal_in_group", "parameterize", "rerun_on_exception", "rerun_if_address_is_in_use", "skip_if_not_enough_gpus", "free_port", "spawn", "clear_cache_before_run", "run_on_environment_flag", "check_state_dict_equal", "assert_hf_output_close", "DummyDataloader", ]