Some syntax edits for build and headers

Author: Jessica Lin

recipes_source/recipes/save_load_across_devices.py

@@ -19,9 +19,11 @@
 first change the runtime to “GPU” or higher. Once you do, we need to
 install ``torch`` if it isn’t already available.
 
-"""
+::
+
+   pip install torch
 
-pip install torch
+"""
 
 
 ######################################################################
@@ -35,7 +37,7 @@
 # 5. Save on a CPU, load on a GPU
 # 6. Saving and loading ``DataParallel`` models
 # 
-# **1) Import necessary libraries for loading our data**
+# 1. Import necessary libraries for loading our data
 # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 # 
 # For this recipe, we will use ``torch`` and its subsidiaries ``torch.nn``
@@ -48,7 +50,7 @@
 
 
 ######################################################################
-# **2) Define and intialize the neural network**
+# 2. Define and intialize the neural network
 # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 # 
 # For sake of example, we will create a neural network for training
@@ -79,7 +81,7 @@ def forward(self, x):
 
 
 ######################################################################
-# **3) Save on GPU, Load on CPU**
+# 3. Save on GPU, Load on CPU
 # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 # 
 # When loading a model on a CPU that was trained with a GPU, pass
@@ -103,7 +105,7 @@ def forward(self, x):
 # In this case, the storages underlying the tensors are dynamically
 # remapped to the CPU device using the ``map_location`` argument.
 # 
-# **4) Save on GPU, Load on GPU**
+# 4. Save on GPU, Load on GPU
 # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 # 
 # When loading a model on a GPU that was trained and saved on GPU, simply
@@ -130,7 +132,7 @@ def forward(self, x):
 # remember to manually overwrite tensors:
 # ``my_tensor = my_tensor.to(torch.device('cuda'))``.
 # 
-# **5) Save on CPU, Load on GPU**
+# 5. Save on CPU, Load on GPU
 # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 # 
 # When loading a model on a GPU that was trained and saved on CPU, set the
@@ -157,7 +159,7 @@ def forward(self, x):
 
 
 ######################################################################
-# **6) Saving ``torch.nn.DataParallel`` Models**
+# 6. Saving ``torch.nn.DataParallel`` Models
 # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 # 
 # ``torch.nn.DataParallel`` is a model wrapper that enables parallel GPU
@@ -185,4 +187,4 @@ def forward(self, x):
 # 
 # -  TBD
 # -  TBD
-# 
+#