mtrencseni/pytorch-basics-7.py

Last active February 8, 2019 21:22

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

Download ZIP

Raw

pytorch-basics-7.py

torch.set_default_tensor_type('torch.cuda.FloatTensor')

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