Trait candle_optimisers::Model

pub trait Model: Sized {
    // Required method
    fn loss(&self) -> CResult<Tensor>;
}

Trait for Models: this is needed for optimisers that require the ability to calculate the loss such as LBFGS

Required Methods

fn loss(&self) -> CResult<Tensor>

get the loss of the model

Object Safety

This trait is not object safe.

Implementors