avalanche.evaluation.metrics.gpu_usage_metrics

avalanche.evaluation.metrics.gpu_usage_metrics(gpu_id, every=0.5, minibatch=False, epoch=False, experience=False, stream=False) List[PluginMetric][source]

Helper method that can be used to obtain the desired set of plugin metrics.

Parameters
  • gpu_id – GPU device ID.

  • every – seconds after which update the maximum GPU usage

  • minibatch – If True, will return a metric able to log the minibatch max GPU usage.

  • epoch – If True, will return a metric able to log the epoch max GPU usage.

  • experience – If True, will return a metric able to log the experience max GPU usage.

  • stream – If True, will return a metric able to log the evaluation max stream GPU usage.

Returns

A list of plugin metrics.