Fix typo RMSProp -> RMSprop (#13063)

This commit is contained in:
Corey Lowman
2020-12-23 16:37:46 -05:00
committed by GitHub
parent 8df94e33e0
commit 668ea0bc26
+1 -1
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"],