Lightning-AI/torchmetrics

Add `_filter_kwargs` in `ClasswiseWrapper` metric wrapper

ioangatop opened this issue · 3 comments

ioangatop commented

As the ClasswiseWrapper accepts agnostic *args, **kwargs arguments, its not handy to use it in a MetricColletion, as tailored input arguments for other metrics get passed in the metrics.

It would be great, just like in MetricColletion, to add the _filter_kwargs for the metric

github-actions commented

Hi! thanks for your contribution!, great first issue!

Borda commented

It would be great, just like in MetricColletion, to add the _filter_kwargs for the metric

sounds fine to me... what do you think @justusschock @SkafteNicki ?

ioangatop commented

Thanks a lot @SkafteNicki @stancld and @Borda 🙏