From 31fa0325fa5654e6838dcd94f0b42850341b08bc Mon Sep 17 00:00:00 2001 From: Matthew Turnshek Date: Thu, 23 Nov 2023 00:05:37 -0500 Subject: [PATCH] Update quickstart argument name (#994) Noticed an improper argument name when going through the quickstart. --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 346f97003..d7587597e 100644 --- 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)