Skip to main content

rnn/optimizers/
kind.rs

1#[derive(Clone, Copy, Debug, PartialEq)]
2pub enum OptimizerKind {
3    Sgd { momentum: f32, nesterov: bool },
4    AdamW { beta1: f32, beta2: f32, eps: f32, weight_decay: f32 },
5}
6
7#[derive(Clone, Copy, Debug, PartialEq, Eq)]
8pub enum OptimizerError {
9    InvalidHyperParams,
10    ShapeMismatch,
11    StepOverflow,
12}