ding.config.example.SAC.gym_bipedalwalker_v3

ding.config.example.SAC.gym_bipedalwalker_v3

Full Source Code

../ding/config/example/SAC/gym_bipedalwalker_v3.py

1from easydict import EasyDict 2import ding.envs.gym_env 3 4cfg = dict( 5 exp_name='BipedalWalker-v3-SAC', 6 seed=0, 7 env=dict( 8 env_id='BipedalWalker-v3', 9 collector_env_num=8, 10 evaluator_env_num=5, 11 n_evaluator_episode=5, 12 act_scale=True, 13 rew_clip=True, 14 ), 15 policy=dict( 16 cuda=True, 17 random_collect_size=10000, 18 model=dict( 19 obs_shape=24, 20 action_shape=4, 21 twin_critic=True, 22 action_space='reparameterization', 23 actor_head_hidden_size=128, 24 critic_head_hidden_size=128, 25 ), 26 learn=dict( 27 update_per_collect=64, 28 batch_size=256, 29 learning_rate_q=0.0003, 30 learning_rate_policy=0.0003, 31 learning_rate_alpha=0.0003, 32 target_theta=0.005, 33 discount_factor=0.99, 34 auto_alpha=True, 35 learner=dict(hook=dict(log_show_after_iter=1000, )) 36 ), 37 collect=dict(n_sample=64, ), 38 other=dict(replay_buffer=dict(replay_buffer_size=300000, ), ), 39 ), 40 wandb_logger=dict( 41 gradient_logger=True, video_logger=True, plot_logger=True, action_logger=True, return_logger=False 42 ), 43) 44 45cfg = EasyDict(cfg) 46 47env = ding.envs.gym_env.env