aihwkit.experiments.runners.metrics module¶
Helper for retrieving Metrics of an Experiment.
- class aihwkit.experiments.runners.metrics.LocalMetric(stdout=True)[source]¶
Bases:
object
Metric for local experiments.
Metric for local execution of experiments. Output to stdout can be controlled by the
stdout
parameter to the constructor.- Parameters
stdout (bool) –
- Return type
None
- receive_train_epoch_batch_end(total, train_loss)[source]¶
Hook for TRAIN_EPOCH_START.
- Parameters
total (int) –
train_loss (float) –
- Return type
None