search

mila-iqia / cookiecutter-pyml

MIT License
19 stars 8 forks source link

Add torchmetrics and gather metrics #79

Closed jerpint closed 1 year ago

jerpint commented 1 year ago

In this PR, we use torchmetrics to add Accuracy, precison, recall, and f1-score, and track them in tensorboard

mirkobronzimila commented 1 year ago

@jerpint : add softmax before calling metrics.

jerpint commented 1 year ago

This is now deprecated. A new version of this appears in #101