mirror of https://github.com/hpcaitech/ColossalAI
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
8 lines
358 B
8 lines
358 B
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)
|
|
|
|
__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'
|
|
]
|