evaluate
torchmetrics
