oa::CbSummary

oa::CbSummary Prints a final training summary at onTrainEnd: loss, wall latency/throughput, GPU mean/p50/p95, the wall-to-GPU timing gap, and total duration. Optionally tracks initial loss for comparison. example output: summary: loss: initial 2.6558 · final 0.2965 · mean 0.4812 Wall: 0.07 ms/step · 943.68K sample/s · 60.40M token/s GPU: mean 0.051 ms/step · p50 0.049 · p95 0.061 · 1.26M sample/s run: 0.32s · 4686 steps · batch 64 · sequence 64 token/sample

ClC++ public headers

Inheritance

public CbTraining · public Callback

Public Methods

void oa::CbSummary::onStepEnd(oa::ItTraining & inIter)

void oa::CbSummary::onTrainEnd(oa::ItTraining & inIter)

void oa::CbSummary::setValidationMetric(const oa::Metric * inMetric)

Examples

ExMl

Learn a 300-token BPE vocabulary, train a causal Transformer for 300 optimizer steps, verify that cross-entropy falls, and generate text from the trained model.

Constructor & Destructor Documentation

oa::CbSummary::CbSummary(
	bool inTrackInitialLoss = true
)

No public source comment is attached to this declaration.

Parameters

inTrackInitialLoss

bool

Default: true

Public Method Documentation

void oa::CbSummary::onStepEnd(
	oa::ItTraining & inIter
)

No public source comment is attached to this declaration.

Parameters

inIter

oa::ItTraining &

—

Returns

void

The declared return value.

void oa::CbSummary::onTrainEnd(
	oa::ItTraining & inIter
)

No public source comment is attached to this declaration.

Parameters

inIter

oa::ItTraining &

—

Returns

void

The declared return value.

void oa::CbSummary::setValidationMetric(
	const oa::Metric * inMetric
)

No public source comment is attached to this declaration.

Parameters

inMetric

const oa::Metric *

—

Returns

void

The declared return value.