update codes

2021-12-28 18:46:52 +08:00
parent 41fb561d25
commit bd51b5a7ad
52 changed files with 305 additions and 292 deletions
--- a/codes/PPO/task0.py
+++ b/codes/PPO/task0.py
@@ -45,9 +45,9 @@ class PlotConfig:
 def env_agent_config(cfg,seed=1):
    env = gym.make(cfg.env_name)  
    env.seed(seed)
-    n_states = env.observation_space.shape[0]
-    n_actions = env.action_space.n
-    agent = PPO(n_states,n_actions,cfg)
+    state_dim = env.observation_space.shape[0]
+    action_dim = env.action_space.n
+    agent = PPO(state_dim,action_dim,cfg)
    return env,agent

 cfg  = PPOConfig()