Torchmetrics
Created: 29 Nov 2022, 03:49 PM | Modified: =dateformat(this.file.mtime,"dd MMM yyyy, hh:mm a")
Tags: knowledge, tools
Implements metrics easily ⇒ no need to write yourself
https://torchmetrics.readthedocs.io/en/stable/pages/quickstart.html
Alternative is torcheval (which has metrics too)
https://github.com/pytorch/torcheval
https://pytorch.org/torcheval/stable/
Also has class methods to get module summary (e.g. number of flops)
See https://github.com/Lightning-AI/lightning/issues/12567#issuecomment-1233546466
And https://pytorch.org/torcheval/stable/torcheval.tools.html