.FluxTraining
AbstractMetric
abstract type
defined in module
FluxTraining
abstract type AbstractMetric
Abstract type for metrics passed to
Metrics
.
For most use cases, you should use
Metric
, the standard implementation.
If
Metric
doesn't fit your use case, you can create a new subtype of
AbstractMetric
and implement the following methods to make it compatible with
Metrics
:
reset!
(metric)
step!
(metric, learner)
stepvalue
(metric)
epochvalue
(metric)
metricname
(metric)