Trait concision_core::nn::models::module::Module

pub trait Module {
    type Config;
    type Params;

    // Required methods
    fn config(&self) -> &Self::Config;
    fn params(&self) -> &Self::Params;
    fn params_mut(&mut self) -> &mut Self::Params;
}

A Module defines any object that may be used as a layer in a neural network. Config is a type that defines the configuration of the module; including any and all hyperparameters. Params is a type that defines the parameters of the module; typically references a Linear set of parameters { weights, bias }

Required Associated Types

type Config

type Params

Required Methods

fn config(&self) -> &Self::Config

fn params(&self) -> &Self::Params

fn params_mut(&mut self) -> &mut Self::Params

Implementors