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.-
receive_train_epoch_batch_end
(total, train_loss)[source]¶ Hook for TRAIN_EPOCH_START.
- Parameters
total (int) –
train_loss (float) –
- Return type
None
-