kd.train.Auxiliaries

kd.train.Auxiliaries#

class kauldron.train.Auxiliaries(
*,
_fake_refs: type[_FakeRefsUnset] | dict[str,
_FakeRootCfg] = <class 'kauldron.utils.config_util._FakeRefsUnset'>,
losses: Mapping[str,
kd_losses.Loss] = _FakeRootCfg('cfg.train_losses'),
metrics: Mapping[str,
kd_metrics.Metric] = _FakeRootCfg('cfg.train_metrics'),
summaries: Mapping[str,
kd_metrics.Metric] = _FakeRootCfg('cfg.train_summaries'),
)[source]

Bases: kauldron.utils.config_util.UpdateFromRootCfg

Wrapper around the losses, summaries and metrics.

losses: Mapping[str, kd_losses.Loss] = _FakeRootCfg('cfg.train_losses')
metrics: Mapping[str, kd_metrics.Metric] = _FakeRootCfg('cfg.train_metrics')
summaries: Mapping[str, kd_metrics.Metric] = _FakeRootCfg('cfg.train_summaries')
update_context(
context: kauldron.train.context.Context,
) kauldron.train.context.Context[source]

Get auxilaries.