Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Conversation

@BanzaiTokyo
Copy link
Contributor

No description provided.

@vfdev-5
Copy link
Collaborator

vfdev-5 commented Mar 28, 2025

Let's also make the following check inside Fbeta implementation that if provided precision, recall metrics then device arg should None, same as this:

if recall is not None and output_transform is not None:
raise ValueError("If recall argument is provided, output_transform should be None")

Also let's change the default value here to None:

device: Union[str, torch.device] = torch.device("cpu"),

If precision, recall are None and device is None then set device to cpu.

OK if you do that in a follow-up PR

@github-actions github-actions bot added the module: metrics Metrics module label Mar 29, 2025
Copy link
Collaborator

@vfdev-5 vfdev-5 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, we can merge as failures are unrelated to the PR

@vfdev-5 vfdev-5 merged commit abd80b8 into pytorch:master Apr 16, 2025
15 of 17 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

module: metrics Metrics module

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants