Skip to content

Instantly share code, notes, and snippets.

@Noob-can-Compile

Noob-can-Compile/Define_loss.py

Created December 1, 2019 12:11
Show Gist options
  • Save Noob-can-Compile/ec76d0d7dd57bd357a9ef8a0a89b254a to your computer and use it in GitHub Desktop.
Save Noob-can-Compile/ec76d0d7dd57bd357a9ef8a0a89b254a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Define_loss.py
## TODO: Define the loss and optimization
import torch.optim as optim
criterion = nn.SmoothL1Loss()
optimizer = optim.Adam(net.parameters(), lr = 0.001)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment