TrainOutput

class braintools.trainer.TrainOutput(loss, metrics=None)

Output container for training step.

Parameters:

• loss (Any) – The loss value to be used for gradient computation.

• metrics (Dict[str, Any] | None) – Additional metrics to log.