From 668ea0bc261d85d4ac2d6474974b02e3ff136bc0 Mon Sep 17 00:00:00 2001 From: Corey Lowman Date: Wed, 23 Dec 2020 16:37:46 -0500 Subject: [PATCH] Fix typo RMSProp -> RMSprop (#13063) --- rllib/agents/impala/vtrace_torch_policy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rllib/agents/impala/vtrace_torch_policy.py b/rllib/agents/impala/vtrace_torch_policy.py index a42707e4d..22077202e 100644 --- a/rllib/agents/impala/vtrace_torch_policy.py +++ b/rllib/agents/impala/vtrace_torch_policy.py @@ -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"],