Trait VisionModel 

pub trait VisionModel: Debug {
    // Required methods
    fn forward(&self, input: &Array4<f64>) -> Result<Array4<f64>>;
    fn parameters(&self) -> &Array1<f64>;
    fn update_parameters(&mut self, params: &Array1<f64>) -> Result<()>;
    fn num_parameters(&self) -> usize;
    fn clone_box(&self) -> Box<dyn VisionModel>;
}

Trait for vision models

Required Methods

fn forward(&self, input: &Array4<f64>) -> Result<Array4<f64>>

Forward pass through the model

fn parameters(&self) -> &Array1<f64>

Get model parameters

fn update_parameters(&mut self, params: &Array1<f64>) -> Result<()>

Update parameters

fn num_parameters(&self) -> usize

Number of parameters

fn clone_box(&self) -> Box<dyn VisionModel>

Clone the model

Trait Implementations

impl Clone for Box<dyn VisionModel>

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

Implementors