Closed jerpint closed 1 year ago
In this PR, we use torchmetrics to add Accuracy, precison, recall, and f1-score, and track them in tensorboard
@jerpint : add softmax before calling metrics.
This is now deprecated. A new version of this appears in #101
In this PR, we use torchmetrics to add Accuracy, precison, recall, and f1-score, and track them in tensorboard