hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 10:31:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix typo RMSProp -> RMSprop (#13063)

Browse source
This commit is contained in:
Corey Lowman 2020-12-23 16:37:46 -05:00 • committed by GitHub
parent 8df94e33e0
commit 668ea0bc26
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
rllib/agents/impala/vtrace_torch_policy.py
View file

@ -246,7 +246,7 @@ def choose_optimizer(policy, config):
return torch.optim.Adam(
params=policy.model.parameters(), lr=policy.cur_lr)
else:
return torch.optim.RMSProp(
return torch.optim.RMSprop(
params=policy.model.parameters(),
lr=policy.cur_lr,
weight_decay=config["decay"],