| from typing import Tuple | |
| from easydict import EasyDict | |
| import sys | |
| import importlib | |
| env_dict = { | |
| 'cartpole': 'dizoo.classic_control.cartpole.config', | |
| 'pendulum': 'dizoo.classic_control.pendulum.config', | |
| } | |
| policy_dict = { | |
| 'dqn': 'ding.policy.dqn', | |
| 'rainbow': 'ding.policy.rainbow', | |
| 'c51': 'ding.policy.c51', | |
| 'qrdqn': 'ding.policy.qrdqn', | |
| 'iqn': 'ding.policy.iqn', | |
| 'a2c': 'ding.policy.a2c', | |
| 'impala': 'ding.policy.impala', | |
| 'ppo': 'ding.policy.ppo', | |
| 'sqn': 'ding.policy.sqn', | |
| 'r2d2': 'ding.policy.r2d2', | |
| 'ddpg': 'ding.policy.ddpg', | |
| 'td3': 'ding.policy.td3', | |
| 'sac': 'ding.policy.sac', | |
| } | |
| def get_predefined_config(env: str, policy: str) -> Tuple[EasyDict, EasyDict]: | |
| config_name = '{}_{}_config'.format(env, policy) | |
| create_config_name = '{}_{}_create_config'.format(env, policy) | |
| try: | |
| m = importlib.import_module(env_dict[env] + '.' + config_name) | |
| return [getattr(m, config_name), getattr(m, create_config_name)] | |
| except ImportError: | |
| print("Please get started by other types, there is no related pre-defined config({})".format(config_name)) | |
| sys.exit(1) | |