Adding `task` param to `torchmetrics.Accuracy` and `torchmetrics.AveragePrecision`

Spijkervet / CLMR

Official PyTorch implementation of Contrastive Learning of Musical Representations

https://arxiv.org/abs/2103.09410

Apache License 2.0

309 stars 48 forks source link

Open fducau opened 1 year ago

fducau commented 1 year ago

This is within clmr.modules.linear_evaluation.LinearEvaluation

Addresses #24