Skip to content

Conversation

@Seilmast
Copy link
Collaborator

Found that some metrics might not have made sure all metrics are on the same device. Made the MetricWrapper force y_true and y_logits onto the cpu

@Seilmast Seilmast added the bug Something isn't working label Feb 24, 2025
Copy link
Contributor

@Johanmkr Johanmkr left a comment

Choose a reason for hiding this comment

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

Perfect!

@Johanmkr Johanmkr merged commit 546bead into main Feb 24, 2025
4 checks passed
@Johanmkr Johanmkr deleted the mag-branch branch February 24, 2025 08:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants