GitHubEmploy/pytorch_4.py

Created December 30, 2021 21:04

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/GitHubEmploy/fd4ffd076221a5d4e33ea6ade2c0b55d.js"></script>
Save GitHubEmploy/fd4ffd076221a5d4e33ea6ade2c0b55d to your computer and use it in GitHub Desktop.

Download ZIP

Raw

pytorch_4.py

	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