ColossalAI/colossalai/context/random/__init__.py

10 lines
356 B
Python
Raw Normal View History

2021-10-28 16:21:23 +00:00
from ._helper import (seed, set_mode, with_seed, add_seed,
get_seeds, get_states, get_current_mode,
2022-01-07 07:08:36 +00:00
set_seed_states, sync_states, moe_set_seed)
2021-10-28 16:21:23 +00:00
__all__ = [
'seed', 'set_mode', 'with_seed', 'add_seed', 'get_seeds',
2022-01-07 07:08:36 +00:00
'get_states', 'get_current_mode', 'set_seed_states', 'sync_states',
'moe_set_seed'
2021-10-28 16:21:23 +00:00
]