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

added Lovasz-loss using pytorch tensors #37

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

gguzzy
Copy link

@gguzzy gguzzy commented Apr 12, 2024

  1. Introduced pytorch tensors to Loss, taking advantage of GPUs in torch.
  2. Added new example test using pytorch tensors

1. Introduced pytorch tensors to Loss, taking advantage of GPUs.
2. Added mew example using pytorch tensors
@gguzzy gguzzy closed this Apr 18, 2024
@gguzzy gguzzy reopened this Apr 18, 2024
@gguzzy
Copy link
Author

gguzzy commented Apr 18, 2024

If you want, I can still arrange in a better way the code, aiming at realizing a custom torch loss, would you like to update the code together or can i progress with it? It should still work, but I wanted to be sure on that, what you think? I think it would be really useful for many tasks, that is why I am working on it. @bermanmaxim

Thanks

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 this pull request may close these issues.

1 participant