ding.config.example.PG.gym_pendulum_v1

ding.config.example.PG.gym_pendulum_v1

Full Source Code

../ding/config/example/PG/gym_pendulum_v1.py

1from easydict import EasyDict 2import ding.envs.gym_env 3 4cfg = dict( 5 exp_name='Pendulum-v1-PG', 6 seed=0, 7 env=dict( 8 env_id='Pendulum-v1', 9 collector_env_num=8, 10 evaluator_env_num=5, 11 n_evaluator_episode=5, 12 stop_value=-200, 13 act_scale=True, 14 ), 15 policy=dict( 16 cuda=False, 17 action_space='continuous', 18 model=dict( 19 action_space='continuous', 20 obs_shape=3, 21 action_shape=1, 22 ), 23 learn=dict( 24 batch_size=4000, 25 learning_rate=0.001, 26 entropy_weight=0.001, 27 ), 28 collect=dict( 29 n_episode=20, 30 unroll_len=1, 31 discount_factor=0.99, 32 ), 33 eval=dict(evaluator=dict(eval_freq=1, )) 34 ), 35 wandb_logger=dict( 36 gradient_logger=True, video_logger=True, plot_logger=True, action_logger=True, return_logger=False 37 ), 38) 39 40cfg = EasyDict(cfg) 41 42env = ding.envs.gym_env.env