update

Author: clee2000

recipes_source/recipes/changing_default_device.py

@@ -35,14 +35,16 @@
     print(mod(torch.randn(128, 20)).device)
 
 #########################################
-# You can also set it globally like this: 
+# You can also set it globally like this:
 
 torch.set_default_device('cuda')
 
 mod = torch.nn.Linear(20, 30)
 print(mod.weight.device)
 print(mod(torch.randn(128, 20)).device)
 
+torch.set_default_device('cpu')
+
 ################################################################
 # This function imposes a slight performance cost on every Python
 # call to the torch API (not just factory functions). If this