burn 0.3.0

BURN: Burn Unstoppable Rusty Neurons
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14

use crate as burn;
use crate::config::Config;

#[derive(Config)]
pub struct AdamConfig {
    /// Learning rate for the optimizer.
    learning_rate: f64,
    /// Parameter for Adam.
    beta_1: f64,
    /// Parameter for Adam.
    beta_2: f64,
    /// A value required for numerical stability, typically 1e-8.
    epsilon: f64,
}