d3rlpy.optimizers.RMSpropFactory

class d3rlpy.optimizers.RMSpropFactory(alpha=0.95, eps=0.01, weight_decay=0, momentum=0, centered=True, **kwargs)[source]

An alias for RMSprop optimizer.

from d3rlpy.optimizers import RMSpropFactory

factory = RMSpropFactory(weight_decay=1e-4)
Parameters:
  • alpha (float) – smoothing constant.
  • eps (float) – term added to the denominator to improve numerical stability.
  • weight_decay (float) – weight decay (L2 penalty).
  • momentum (float) – momentum factor.
  • centered (bool) – flag to compute the centered RMSProp, the gradient is normalized by an estimation of its variance.
optim_cls

torch.optim.RMSprop class.

Type:type
optim_kwargs

given parameters for an optimizer.

Type:dict

Methods

create(params, lr)

Returns an optimizer object.

Parameters:
  • params (list) – a list of PyTorch parameters.
  • lr (float) – learning rate.
Returns:

an optimizer object.
Return type:

torch.optim.Optimizer
get_params(deep=False)

Returns optimizer parameters.

Parameters:deep (bool) – flag to deeply copy the parameters.
Returns:optimizer parameters.
Return type:dict