LoicGrobol / zeldarose

Train transformer-based models.
https://zeldarose.readthedocs.io
Other
28 stars 3 forks source link

Use TorchMetrics BinaryAccuracy directly #57

Open LoicGrobol opened 1 year ago

LoicGrobol commented 1 year ago

It needs a bit of work on our part but it avoids having to make our own metric and go too deep into the lib API. See https://torchmetrics.readthedocs.io/en/stable/classification/accuracy.html#binaryaccuracy