ColossalAI/colossalai/context/random/__init__.py

8 lines
358 B
Python
Raw Normal View History

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