Skip to content

Instantly share code, notes, and snippets.

@SannaPersson
Created March 21, 2021 10:47
Show Gist options
  • Save SannaPersson/c4e230917cebda8f48157414409374e2 to your computer and use it in GitHub Desktop.
Save SannaPersson/c4e230917cebda8f48157414409374e2 to your computer and use it in GitHub Desktop.
class_loss = self.entropy(predictions[..., 5:][obj], target[..., 5][obj].long()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment