Skip to content

Instantly share code, notes, and snippets.

@omarsar

omarsar/loss_optimizer.py

Created December 22, 2019 14:11
Show Gist options
  • Save omarsar/152eaf63c9df1d95f5fd124c6bd43044 to your computer and use it in GitHub Desktop.
Save omarsar/152eaf63c9df1d95f5fd124c6bd43044 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
loss_optimizer.py
## loss function
criterion = nn.MSELoss()
## optimizer algorithm
optimizer = torch.optim.SGD(model.parameters(), lr=0.01)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment