Skip to content

Instantly share code, notes, and snippets.

@NilakshanKunananthaseelan

NilakshanKunananthaseelan/optim.py

Last active April 22, 2022 07:36
Show Gist options
  • Save NilakshanKunananthaseelan/bd4eb044b8bfe01026d6a5892375b2f1 to your computer and use it in GitHub Desktop.
Save NilakshanKunananthaseelan/bd4eb044b8bfe01026d6a5892375b2f1 to your computer and use it in GitHub Desktop.
Download ZIP
Optimizer and criterion
Raw
optim.py
import torch.optim as optim
net = Net()
criterion = nn.MSELoss()
optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.0,weight_decay=0.000)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment