vinimonteiro/nn_pytorch_playing.py

Created December 17, 2021 15:02

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

Download ZIP

nn_pytorch_playing

Raw

nn_pytorch_playing.py

	plt.imshow(X[1].view(28,28))
	plt.show()

	print(torch.argmax(model(X[1].view(-1, 784))[0]))

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