viva_tensor/examples/training

Types

pub type TrainingState {
  TrainingState(
    tape: autograd.Tape,
    layer1: layers.Linear,
    layer2: layers.Linear,
  )
}

Constructors

Values

pub fn main() -> Nil
Search Document