~/jendhub

def train_at_loss

bendtorch@1.3.0.0 · main.bend:960

import bendtorch@1.3.0.0/main.bend as Bendtorch
import Base

960def train_at_loss(-P: Type, -X: Type, -Y: Type, -S: Type, bf: P -> S -> Y -> P -> Bwd<P, X>, sf: Opt -> P -> P -> OptState<P> -> Step<P>, +cfg: OptCfg, p: P, s: S, g: P, st: OptState<P>, +p1: F32, +p2: F32, r: LossOut<Y>) -> Train<P>:
961  match r:
962    case LossOut{l, dy}: train_at_b(P, X, sf, cfg, st, p1, p2, l, bf(p, s, dy, g))

Uses (9)

Used by (1)