from torchmetrics.aggregation import MaxMetric metric = MaxMetric() values = [ ] for i in range(10): values.append(metric(i)) fig_, ax_ = metric.plot(values)