Cannot use average method 'none' in Dice meter #2262
Labels
bug / fix
Something isn't working
good first issue
Good for newcomers
help wanted
Extra attention is needed
v1.2.x
🐛 Bug
In the docstring of
it states that
average
allowsnone
or None input wherein a class-wise result will be reported. However the code here https://github.com/Lightning-AI/torchmetrics/blob/master/src/torchmetrics/classification/dice.py#L176 suggests that a ValueError will be raised and thereforeaverage
must be micro/macro/sample.It appears that the value check is redundant considering the following line was added already.
To Reproduce
Steps to reproduce the behavior...
Code sample
Expected behavior
Environment
conda
,pip
, build from source):Additional context
The text was updated successfully, but these errors were encountered: