How to read model weights using FluxTraining (stateaccess issues)?

Specific Domains Machine Learning
,
4

The issue is that stateaccess need to be for ::Metrics not ::MyMetric.

function FluxTraining.stateaccess(::FluxTraining.Metrics)
    (model = FluxTraining.Read(),
        params = FluxTraining.Read(),
        cbstate = (metricsstep = FluxTraining.Write(),
            metricsepoch = FluxTraining.Write(),
            history = FluxTraining.Read()),
        step = FluxTraining.Read())
end