mirror of
https://github.com/wassname/ray.git
synced 2026-06-27 21:53:18 +08:00
Fix typo RMSProp -> RMSprop (#13063)
This commit is contained in:
@@ -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"],
|
||||
|
||||
Reference in New Issue
Block a user