Skip to content

Instantly share code, notes, and snippets.

@dmesquita

dmesquita/optimizer.py

Created October 25, 2017 12:13
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save dmesquita/f6aa895690270e1256dab547afb99193 to your computer and use it in GitHub Desktop.
Save dmesquita/f6aa895690270e1256dab547afb99193 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
optimizer.py
net = OurNet(input_size, hidden_size, num_classes)
criterion = nn.CrossEntropyLoss()
optimizer = torch.optim.Adam(net.parameters(), lr=learning_rate)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment