spauq.wrapper.torchmetrics