native_neural_network 0.3.1

Lib no_std Rust for native neural network (.rnn)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

#[derive(Clone, Copy, Debug, PartialEq)]
pub enum LrSchedule {
    Constant,
    StepDecay { step_size: u32, gamma: f32 },
    Cosine { total_steps: u32, min_lr_ratio: f32 },
    LinearWarmup { warmup_steps: u32 },
}

pub enum LrScheduleF64 {
    Constant,
    StepDecay { step_size: u32, gamma: f64 },
    Cosine { total_steps: u32, min_lr_ratio: f64 },
    LinearWarmup { warmup_steps: u32 },
}

#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub enum ScheduleError {
    InvalidConfig,
}