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