Skip to content

Instantly share code, notes, and snippets.

@albertlai431
Created December 4, 2018 00:33
Show Gist options
  • Save albertlai431/708e096b1f0caab3a035ce6789aeb0fc to your computer and use it in GitHub Desktop.
Save albertlai431/708e096b1f0caab3a035ce6789aeb0fc to your computer and use it in GitHub Desktop.
Instance of Optimizer
#Create instance of optimizer (Adam)
optimizer = torch.optim.Adam(model.parameters(), lr=learning_rate)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment