optim#


class OptimizerFactory[source]#
create_optimizer(module: Module, lr: float) Optimizer[source]#
abstract create_optimizer_for_params(params: Iterable[Tensor] | Iterable[dict[str, Any]], lr: float) Optimizer[source]#
class OptimizerFactoryAdam(betas: tuple[float, float] = (0.9, 0.999), eps: float = 1e-08, weight_decay: float = 0)[source]#
create_optimizer_for_params(params: Iterable[Tensor] | Iterable[dict[str, Any]], lr: float) Optimizer[source]#
class OptimizerFactoryRMSprop(alpha: float = 0.99, eps: float = 1e-08, weight_decay: float = 0, momentum: float = 0, centered: bool = False)[source]#
create_optimizer_for_params(params: Iterable[Tensor] | Iterable[dict[str, Any]], lr: float) Optimizer[source]#
class OptimizerFactoryTorch(optim_class: OptimizerWithLearningRateProtocol, **kwargs: Any)[source]#
create_optimizer_for_params(params: Iterable[Tensor] | Iterable[dict[str, Any]], lr: float) Optimizer[source]#
class OptimizerWithLearningRateProtocol(*args, **kwargs)[source]#