Contents

struct

SushiAI::Train::StepSpec

The four graph values a training loop has to know by name.

Declared in
include/SushiAI/train/trainer.hpp

Public attributes

Graph::ValueId inputs = Graph::NO_VALUE

The [N, D] minibatch input leaf, rewritten per step.

Graph::ValueId targets = Graph::NO_VALUE

The [N, C] one-hot target leaf, rewritten per step.

Graph::ValueId logits = Graph::NO_VALUE

The model's [N, C] logits, read for accuracy.

Graph::ValueId loss = Graph::NO_VALUE

The 1-element loss that was differentiated.