Skip to content

Instantly share code, notes, and snippets.

@GitHubEmploy
Created December 30, 2021 21:04
Show Gist options
  • Save GitHubEmploy/fd4ffd076221a5d4e33ea6ade2c0b55d to your computer and use it in GitHub Desktop.
Save GitHubEmploy/fd4ffd076221a5d4e33ea6ade2c0b55d to your computer and use it in GitHub Desktop.
optimizer = torch.optim.Adam(model.parameters(), lr=0.001)
loss_function = nn.CrossEntropyLoss()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment