Skip to content

Instantly share code, notes, and snippets.

@danielcwq

danielcwq/mnist_loss_func.py

Created November 28, 2021 05:48
Show Gist options
  • Save danielcwq/f8f9008af8455c4a5693d63edbbc8fde to your computer and use it in GitHub Desktop.
Save danielcwq/f8f9008af8455c4a5693d63edbbc8fde to your computer and use it in GitHub Desktop.
Download ZIP
Creating the mnist_loss function
Raw
mnist_loss_func.py
def mnist_loss(predictions, targets):
return torch.where(targets==1, 1-predictions, predictions).mean()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment