Function example_rnn_regression::train

source ·

pub(crate) fn train<Framework: IFramework + 'static>(
    backend: Rc<Backend<Framework>>,
    batch_size: usize,
    learning_rate: f32,
    momentum: f32,
    file: &PathBuf
)where
    Backend<Framework>: IBackend + SolverOps<f32> + LayerOps<f32>,

Expand description

Train, and optionally, save the resulting network state/weights