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]

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.