shangeth/alex.py

Created December 27, 2018 05:56

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

Download ZIP

Raw

alex.py

	x = torch.randn(1, 3, 224, 224).uniform_(0, 1)
	alexnet = AlexNet()
	alexnet(x).size()

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