update codes

2021-12-22 16:55:09 +08:00
parent 75df999258
commit 41fb561d25
75 changed files with 1248 additions and 918 deletions
--- a/codes/PPO/train.py
+++ b/codes/PPO/train.py
@@ -99,9 +99,9 @@ if __name__ == '__main__':
    def env_agent_config(cfg,seed=1):
        env = gym.make(cfg.env_name)  
        env.seed(seed)
-        state_dim = env.observation_space.shape[0]
-        action_dim = env.action_space.n
-        agent = PPO(state_dim,action_dim,cfg)
+        n_states = env.observation_space.shape[0]
+        n_actions = env.action_space.n
+        agent = PPO(n_states,n_actions,cfg)
        return env,agent

    cfg  = PPOConfig()