Fix pytorch#233

Author: chsasank

beginner_source/examples_autograd/two_layer_net_autograd.py

@@ -18,7 +18,7 @@
 
 dtype = torch.float
 device = torch.device("cpu")
-# dtype = torch.device("cuda:0") # Uncomment this to run on GPU
+# device = torch.device("cuda:0") # Uncomment this to run on GPU
 
 # N is batch size; D_in is input dimension;
 # H is hidden dimension; D_out is output dimension.

beginner_source/examples_autograd/two_layer_net_custom_function.py

@@ -48,7 +48,7 @@ def backward(ctx, grad_output):
 
 dtype = torch.float
 device = torch.device("cpu")
-# dtype = torch.device("cuda:0") # Uncomment this to run on GPU
+# device = torch.device("cuda:0") # Uncomment this to run on GPU
 
 # N is batch size; D_in is input dimension;
 # H is hidden dimension; D_out is output dimension.

beginner_source/examples_tensor/two_layer_net_tensor.py

@@ -23,7 +23,7 @@
 
 dtype = torch.float
 device = torch.device("cpu")
-# dtype = torch.device("cuda:0") # Uncomment this to run on GPU
+# device = torch.device("cuda:0") # Uncomment this to run on GPU
 
 # N is batch size; D_in is input dimension;
 # H is hidden dimension; D_out is output dimension.