Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add support for PolyLoss in torchvision #6439

Open
frgfm opened this issue Aug 17, 2022 · 0 comments · May be fixed by #6457
Open

Add support for PolyLoss in torchvision #6439

frgfm opened this issue Aug 17, 2022 · 0 comments · May be fixed by #6457
Assignees

Comments

@frgfm
Copy link
Contributor

frgfm commented Aug 17, 2022

🚀 The feature

The goal would be to add a PolyLoss PyTorch module behaving exactly the same way as any PyTorch core loss module. And the implementation would follow the following paper: https://arxiv.org/abs/2204.12511

One question still pending though: where should this be located in torchvision?

Motivation, pitch

As discussed in #6323, some losses have popped up over the last year. Before they can make their way to core, we should integrate those in torchvision first.

Alternatives

No response

Additional context

No response

@frgfm frgfm linked a pull request Aug 20, 2022 that will close this issue
2 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant