d3rlpy.models.optimizers.RMSpropFactory

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

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.

• kwargs (Any) –

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[str, Any]