Struct burn::train::metric::CudaMetric

pub struct CudaMetric { /* private fields */ }

Track basic cuda infos.

Implementations

impl CudaMetric

pub fn new() -> CudaMetric

Creates a new metric for CUDA.

Trait Implementations

impl Default for CudaMetric

fn default() -> CudaMetric

Returns the “default value” for a type.

impl Metric for CudaMetric

const NAME: &'static str = "CUDA Stats"

The name of the metric.

type Input = ()

The input type of the metric.

fn update(&mut self, _item: &(), _metadata: &MetricMetadata) -> MetricEntry

Update the metric state and returns the current metric entry.

fn clear(&mut self)

Clear the metric state.