Skip to content

Instantly share code, notes, and snippets.

@kabirahuja2431

kabirahuja2431/loss_opti.py

Created October 9, 2019 14:47
Show Gist options
  • Save kabirahuja2431/142ed9023af787cf07c9f53133c48ece to your computer and use it in GitHub Desktop.
Save kabirahuja2431/142ed9023af787cf07c9f53133c48ece to your computer and use it in GitHub Desktop.
Download ZIP
Raw
loss_opti.py
import torch.nn as nn
import torch.optim as optim
criterion = nn.BCEWithLogitsLoss()
opti = optim.Adam(net.parameters(), lr = 2e-5)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment