aihwkit.experiments.runners.i_metrics module

Helper for retrieving Metrics of an Experiment.

class aihwkit.experiments.runners.i_metrics.InferenceLocalMetric(stdout=False)[source]

Bases: object

Metric used by the InferenceWorker Runner.

Parameters:

stdout (bool) –

receive_repeat_end(t_inference_array, avg_acc_arr, std_acc_arr, avg_err_arr, avg_loss_arr, inference_repeats)[source]

Hook for INFERENCE_REPEAT_END.

Parameters:

  • t_inference_array (list) –

  • avg_acc_arr (list) –

  • std_acc_arr (list) –

  • avg_err_arr (list) –

  • avg_loss_arr (list) –

  • inference_repeats (int) –

Return type:

Dict

receive_repeat_start(repeat)[source]

Hook for INFERENCE_REPEAT_START.

Parameters:

repeat (int) –

Return type:

None