type Train

bendtorch@1.3.0.0 · main.bend:901

import Base

901# Parameters, gradient (zero between steps), optimizer-specific buffers, Adam's 1 - b1^t and 1 - b2^t, last loss.
902type Train<-P: Type> is Type:
903  Train{p: P, g: P, st: OptState<P>, p1: F32, p2: F32, loss: F32}