torchmetrics-blog. Code for "TorchMetrics -- How do we use it, and what's the difference between .update() and .forward()"