ding.utils.registry_factory

ding.utils.registry_factory

Full Source Code

../ding/utils/registry_factory.py

1from .registry import Registry 2 3POLICY_REGISTRY = Registry() 4ENV_REGISTRY = Registry() 5ENV_WRAPPER_REGISTRY = Registry() 6LEARNER_REGISTRY = Registry() 7COMM_LEARNER_REGISTRY = Registry() 8SERIAL_COLLECTOR_REGISTRY = Registry() 9PARALLEL_COLLECTOR_REGISTRY = Registry() 10COMM_COLLECTOR_REGISTRY = Registry() 11BUFFER_REGISTRY = Registry() 12COMMANDER_REGISTRY = Registry() 13LEAGUE_REGISTRY = Registry() 14PLAYER_REGISTRY = Registry() 15MODEL_REGISTRY = Registry() 16ENV_MANAGER_REGISTRY = Registry() 17REWARD_MODEL_REGISTRY = Registry() 18DATASET_REGISTRY = Registry() 19SERIAL_EVALUATOR_REGISTRY = Registry() 20MQ_REGISTRY = Registry() 21WORLD_MODEL_REGISTRY = Registry() 22STOCHASTIC_OPTIMIZER_REGISTRY = Registry() 23 24registries = { 25 'policy': POLICY_REGISTRY, 26 'env': ENV_REGISTRY, 27 'env_wrapper': ENV_WRAPPER_REGISTRY, 28 'model': MODEL_REGISTRY, 29 'reward_model': REWARD_MODEL_REGISTRY, 30 'learner': LEARNER_REGISTRY, 31 'serial_collector': SERIAL_COLLECTOR_REGISTRY, 32 'parallel_collector': PARALLEL_COLLECTOR_REGISTRY, 33 'env_manager': ENV_MANAGER_REGISTRY, 34 'comm_learner': COMM_LEARNER_REGISTRY, 35 'comm_collector': COMM_COLLECTOR_REGISTRY, 36 'commander': COMMANDER_REGISTRY, 37 'league': LEAGUE_REGISTRY, 38 'player': PLAYER_REGISTRY, 39 'buffer': BUFFER_REGISTRY, 40 'dataset': DATASET_REGISTRY, 41 'serial_evaluator': SERIAL_EVALUATOR_REGISTRY, 42 'message_queue': MQ_REGISTRY, 43 'world_model': WORLD_MODEL_REGISTRY, 44 'stochastic_optimizer': STOCHASTIC_OPTIMIZER_REGISTRY, 45}