ding.config.example.TD3.gym_lunarlandercontinuous_v2

ding.config.example.TD3.gym_lunarlandercontinuous_v2

Full Source Code

../ding/config/example/TD3/gym_lunarlandercontinuous_v2.py

1from easydict import EasyDict 2from functools import partial 3import ding.envs.gym_env 4 5cfg = dict( 6 exp_name='LunarLanderContinuous-V2-TD3', 7 seed=0, 8 env=dict( 9 env_id='LunarLanderContinuous-v2', 10 collector_env_num=4, 11 evaluator_env_num=8, 12 n_evaluator_episode=8, 13 stop_value=240, 14 act_scale=True, 15 ), 16 policy=dict( 17 cuda=True, 18 random_collect_size=10000, 19 model=dict( 20 obs_shape=8, 21 action_shape=2, 22 action_space='regression', 23 ), 24 learn=dict( 25 update_per_collect=256, 26 batch_size=256, 27 learning_rate_actor=3e-4, 28 learning_rate_critic=1e-3, 29 noise=True, 30 noise_sigma=0.1, 31 noise_range=dict( 32 min=-0.5, 33 max=0.5, 34 ), 35 ), 36 collect=dict( 37 n_sample=256, 38 noise_sigma=0.1, 39 ), 40 eval=dict(evaluator=dict(eval_freq=1000, ), ), 41 other=dict(replay_buffer=dict(replay_buffer_size=100000, ), ), 42 ), 43 wandb_logger=dict( 44 gradient_logger=True, video_logger=True, plot_logger=True, action_logger=True, return_logger=False 45 ), 46) 47 48cfg = EasyDict(cfg) 49 50env = partial(ding.envs.gym_env.env, continuous=True)