Skip to content

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