hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update quickstart argument name (#994)

Browse Source 
Noticed an improper argument name when going through the quickstart.
This commit is contained in:
Matthew Turnshek 2023-11-23 00:05:37 -05:00 committed by GitHub
parent 3a1bc18add
commit 31fa0325fa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@ -240,9 +240,9 @@ Setup policy and collectors:
policy = ts.policy.DQNPolicy( policy = ts.policy.DQNPolicy(
model=net, model=net,
optim=optim, optim=optim,
gamma=gamma, discount_factor=gamma,
action_space=env.action_space, action_space=env.action_space,
estimate_space=n_step, estimation_step=n_step,
target_update_freq=target_freq target_update_freq=target_freq
) )
train_collector = ts.data.Collector(policy, train_envs, ts.data.VectorReplayBuffer(buffer_size, train_num), exploration_noise=True) train_collector = ts.data.Collector(policy, train_envs, ts.data.VectorReplayBuffer(buffer_size, train_num), exploration_noise=True)