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/DQN/task2.py
+++ b/codes/DQN/task2.py
@@ -68,9 +68,9 @@ def env_agent_config(cfg, seed=1):
    # env    = wrap_deepmind(env)
    # env    = wrap_pytorch(env) 
    env.seed(seed)  # 设置随机种子
-    n_states = env.observation_space.shape[0]  # 状态数
-    n_actions = env.action_space.n  # 动作数
-    agent = DQN(n_states, n_actions, cfg)  # 创建智能体
+    state_dim = env.observation_space.shape[0]  # 状态维度
+    action_dim = env.action_space.n  # 动作维度
+    agent = DQN(state_dim, action_dim, cfg)  # 创建智能体
    return env, agent

 def train(cfg, env, agent):