Viewing a single comment thread. View all comments

donshell t1_j9spgvp wrote on February 24, 2023 at 7:36 AM

I think you are looking for TorchMetrics: https://torchmetrics.readthedocs.io/

PHEEEEELLLLLEEEEP t1_j9wnx8o wrote on February 25, 2023 at 2:22 AM

>I tried the torchmetrics implementation, however they’re giving me completely wrong results

Same, the torchmetrics implementation is completely off