Skip to content

ding.config.example.SAC.gym_lunarlandercontinuous_v2

ding.config.example.SAC.gym_lunarlandercontinuous_v2

Full Source Code

../ding/config/example/SAC/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-SAC', 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=260, 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='reparameterization', 23 twin_critic=True, 24 ), 25 learn=dict( 26 update_per_collect=256, 27 batch_size=128, 28 learning_rate_q=1e-3, 29 learning_rate_policy=3e-4, 30 learning_rate_alpha=3e-4, 31 auto_alpha=True, 32 ), 33 collect=dict(n_sample=256, ), 34 eval=dict(evaluator=dict(eval_freq=1000, ), ), 35 other=dict(replay_buffer=dict(replay_buffer_size=int(1e5), ), ), 36 ), 37 wandb_logger=dict( 38 gradient_logger=True, video_logger=True, plot_logger=True, action_logger=True, return_logger=False 39 ), 40) 41 42cfg = EasyDict(cfg) 43 44env = partial(ding.envs.gym_env.env, continuous=True)