optim

Source code: tianshou/highlevel/optim.py

---

class OptimizerFactory

abstract create_optimizer(module: Module, lr: float) → Optimizer

class OptimizerFactoryAdam(betas: tuple[float, float] = (0.9, 0.999), eps: float = 1e-08, weight_decay: float = 0)

create_optimizer(module: Module, lr: float) → Adam

class OptimizerFactoryRMSprop(alpha: float = 0.99, eps: float = 1e-08, weight_decay: float = 0, momentum: float = 0, centered: bool = False)

create_optimizer(module: Module, lr: float) → RMSprop

class OptimizerFactoryTorch(optim_class: OptimizerWithLearningRateProtocol, **kwargs: Any)

create_optimizer(module: Module, lr: float) → Optimizer

class OptimizerWithLearningRateProtocol(*args, **kwargs)