ihoromi4/get_device_method.py

Last active February 5, 2020 11:50

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

Download ZIP

pytorch - get nn.Module device method

Raw

get_device_method.py

	from torch import nn


	class NN(nn.Module):
	def device(self):
	return next(self.parameters()).device

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