ding.config.example.C51.gym_lunarlander_v2

ding.config.example.C51.gym_lunarlander_v2

Full Source Code

../ding/config/example/C51/gym_lunarlander_v2.py

1from easydict import EasyDict 2import ding.envs.gym_env 3 4cfg = dict( 5 exp_name='lunarlander_c51', 6 seed=0, 7 env=dict( 8 collector_env_num=8, 9 evaluator_env_num=8, 10 env_id='LunarLander-v2', 11 n_evaluator_episode=8, 12 stop_value=260, 13 ), 14 policy=dict( 15 cuda=False, 16 model=dict( 17 obs_shape=8, 18 action_shape=4, 19 encoder_hidden_size_list=[512, 64], 20 v_min=-30, 21 v_max=30, 22 n_atom=51, 23 ), 24 discount_factor=0.99, 25 nstep=3, 26 learn=dict( 27 update_per_collect=10, 28 batch_size=64, 29 learning_rate=0.001, 30 target_update_freq=100, 31 ), 32 collect=dict( 33 n_sample=64, 34 unroll_len=1, 35 ), 36 other=dict( 37 eps=dict( 38 type='exp', 39 start=0.95, 40 end=0.1, 41 decay=50000, 42 ), replay_buffer=dict(replay_buffer_size=100000, ) 43 ), 44 ), 45 wandb_logger=dict( 46 gradient_logger=True, video_logger=True, plot_logger=True, action_logger=True, return_logger=False 47 ), 48) 49 50cfg = EasyDict(cfg) 51 52env = ding.envs.gym_env.env