Skip to content

ding.config.example.TD3.gym_hopper_v3

ding.config.example.TD3.gym_hopper_v3

Full Source Code

../ding/config/example/TD3/gym_hopper_v3.py

1from easydict import EasyDict 2import ding.envs.gym_env 3 4cfg = dict( 5 exp_name='Hopper-v3-TD3', 6 seed=0, 7 env=dict( 8 env_id='Hopper-v3', 9 collector_env_num=8, 10 evaluator_env_num=8, 11 n_evaluator_episode=8, 12 stop_value=6000, 13 env_wrapper='mujoco_default', 14 ), 15 policy=dict( 16 cuda=True, 17 random_collect_size=25000, 18 model=dict( 19 obs_shape=11, 20 action_shape=3, 21 actor_head_hidden_size=256, 22 critic_head_hidden_size=256, 23 action_space='regression', 24 ), 25 collect=dict(n_sample=1, ), 26 other=dict(replay_buffer=dict(replay_buffer_size=1000000, ), ), 27 ), 28 wandb_logger=dict( 29 gradient_logger=True, video_logger=True, plot_logger=True, action_logger=True, return_logger=False 30 ), 31) 32 33cfg = EasyDict(cfg) 34 35env = ding.envs.gym_env.env