Skip to content

Latest commit

 

History

History
42 lines (28 loc) · 1.05 KB

optimizers.rst

File metadata and controls

42 lines (28 loc) · 1.05 KB
Raw

Optimizers

d3rlpy.models.optimizers

d3rlpy provides OptimizerFactory that gives you flexible control over optimizers. OptimizerFactory takes PyTorch's optimizer class and its arguments to initialize, which you can check more here.

from torch.optim import Adam
from d3rlpy.algos import DQN
from d3rlpy.models.optimizers import OptimizerFactory

# modify weight decay
optim_factory = OptimizerFactory(Adam, weight_decay=1e-4)

# set OptimizerFactory
dqn = DQN(optim_factory=optim_factory)

There are also convenient alises.

from d3rlpy.models.optimizers import AdamFactory

# alias for Adam optimizer
optim_factory = AdamFactory(weight_decay=1e-4)

dqn = DQN(optim_factory=optim_factory)

d3rlpy.models.optimizers.OptimizerFactory d3rlpy.models.optimizers.SGDFactory d3rlpy.models.optimizers.AdamFactory d3rlpy.models.optimizers.RMSpropFactory