moving input data to device (#487)

ShishirPatil · soumith · commit d6aa1f08ca5d · 2019-05-27T02:52:27.000-04:00
diff --git a/beginner_source/blitz/cifar10_tutorial.py b/beginner_source/blitz/cifar10_tutorial.py
@@ -160,7 +160,7 @@ def forward(self, x):
 
     running_loss = 0.0
     for i, data in enumerate(trainloader, 0):
-        # get the inputs
+        # get the inputs; data is a list of [inputs, labels]
         inputs, labels = data
 
         # zero the parameter gradients
@@ -295,7 +295,7 @@ def forward(self, x):
 #
 # .. code:: python
 #
-#         inputs, labels = inputs.to(device), labels.to(device)
+#         inputs, labels = data[0].to(device), data[1].to(device)
 #
 # Why dont I notice MASSIVE speedup compared to CPU? Because your network
 # is realllly small.

Original file line number	Diff line number	Diff line change
`@@ -160,7 +160,7 @@ def forward(self, x):`
`160`	`160`
`161`	`161`	`running_loss = 0.0`
`162`	`162`	`for i, data in enumerate(trainloader, 0):`
`163`		`- # get the inputs`
	`163`	`+ # get the inputs; data is a list of [inputs, labels]`
`164`	`164`	`inputs, labels = data`
`165`	`165`
`166`	`166`	`# zero the parameter gradients`
`@@ -295,7 +295,7 @@ def forward(self, x):`
`295`	`295`	`#`
`296`	`296`	`# .. code:: python`
`297`	`297`	`#`
`298`		`-# inputs, labels = inputs.to(device), labels.to(device)`
	`298`	`+# inputs, labels = data[0].to(device), data[1].to(device)`
`299`	`299`	`#`
`300`	`300`	`# Why dont I notice MASSIVE speedup compared to CPU? Because your network`
`301`	`301`	`# is realllly small.`