kabirahuja2431/loss_opti.py

Created October 9, 2019 14:47

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/kabirahuja2431/142ed9023af787cf07c9f53133c48ece.js"></script>
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