Update quickstart argument name (#994)

Noticed an improper argument name when going through the quickstart.
thu-ml · Nov 23, 2023 · 31fa032 · 31fa032
1 parent 3a1bc18
commit 31fa032
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -240,9 +240,9 @@ Setup policy and collectors:
 policy = ts.policy.DQNPolicy(
     model=net,
     optim=optim,
-    gamma=gamma, 
-    action_space=env.action_space, 
-    estimate_space=n_step,
+    discount_factor=gamma, 
+    action_space=env.action_space,
+    estimation_step=n_step,
     target_update_freq=target_freq
 )
 train_collector = ts.data.Collector(policy, train_envs, ts.data.VectorReplayBuffer(buffer_size, train_num), exploration_noise=True)