d3rlpy.models.optimizers.RMSpropFactory¶
-
class
d3rlpy.models.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 – smoothing constant.
eps – term added to the denominator to improve numerical stability.
weight_decay – weight decay (L2 penalty).
momentum – momentum factor.
centered – flag to compute the centered RMSProp, the gradient is normalized by an estimation of its variance.
Methods
-
create
(params, lr)¶ Returns an optimizer object.