adamoudad/pytorch_build_model.py

Created March 3, 2021 15:59

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

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/adamoudad/e0ac5512b1493ecc24ccd2e0ea5f689d.js"></script>
Save adamoudad/e0ac5512b1493ecc24ccd2e0ea5f689d to your computer and use it in GitHub Desktop.

Download ZIP

Raw

pytorch_build_model.py

	import torch
	criterion = nn.BCELoss()
	optimizer = torch.optim.Adam(model.parameters())
	model = CustomModel()

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment