Skip to content
Discussion options

You must be logged in to vote

Setting average=None in TorchMetrics’ Dice metric should return per-class scores as per the docs. If this causes errors, it’s likely due to an outdated version or input format issues.
Make sure to update to the latest TorchMetrics version via:

pip install --upgrade torchmetrics

Check input shapes and types match expected formats. Rewriting Dice is usually unnecessary since average=None is supported to get class-wise metrics. If problems persist, consider opening an issue with a reproducible example for help.

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by SkafteNicki
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants