Update README.md

Super minor issue, but caught my eye :)

Author: akiani

README.md

@@ -440,7 +440,7 @@ In this example we use the `nn` package to implement our two-layer network:
 import torch
 
 device = torch.device('cpu')
-device = torch.device('cuda') # Uncomment this to run on GPU
+# device = torch.device('cuda') # Uncomment this to run on GPU
 
 # N is batch size; D_in is input dimension;
 # H is hidden dimension; D_out is output dimension.