Module relearn::torch::optimizers

Optimizers

Structs

AdamConfig

Configuration for the Adam optimizer.

AdamWConfig

Configuration for the AdamW optimizer.

ConjugateGradientOptimizer

Conjugate gradient optimizer

ConjugateGradientOptimizerConfig

Configuration for the Conjugate Gradient Optimizer

RmsPropConfig

Configuration for the RMSProp optimizer.

SgdConfig

Configuration for the SGD optimizer.

Enums

OptimizerStepError

Error performing an optimization step.

Traits

BaseOptimizer

Base optimizer interface

BuildOptimizer

Build an optimizer

Optimizer

Optimizer that minimizes a loss function.

TrustRegionOptimizer

Optimizer that minimizes a loss function subject to a trust region constraint on each step.

Functions

opt_expect_ok_log

Convert optimizer result into Option<T> with logging or panic on error.