viva_tensor/examples/training
Types
pub type TrainingState {
TrainingState(
tape: autograd.Tape,
layer1: layers.Linear,
layer2: layers.Linear,
)
}
Constructors
-
TrainingState( tape: autograd.Tape, layer1: layers.Linear, layer2: layers.Linear, )