d3rlpy.models.optimizers.OptimizerFactory¶
-
class
d3rlpy.models.optimizers.
OptimizerFactory
(optim_cls, **kwargs)[source]¶ A factory class that creates an optimizer object in a lazy way.
The optimizers in algorithms can be configured through this factory class.
from torch.optim Adam from d3rlpy.optimizers import OptimizerFactory from d3rlpy.algos import DQN factory = OptimizerFactory(Adam, eps=0.001) dqn = DQN(optim_factory=factory)
- Parameters
optim_cls – An optimizer class.
kwargs – arbitrary keyword-arguments.
Methods