diff --git a/beginner_source/introyt/trainingyt.py b/beginner_source/introyt/trainingyt.py
index 5c374ca7a37..6b6bf8a8b80 100644
--- a/beginner_source/introyt/trainingyt.py
+++ b/beginner_source/introyt/trainingyt.py
@@ -321,12 +321,12 @@ def train_one_epoch(epoch_index, tb_writer):
 
 #########################################################################
 # To load a saved version of the model:
-# 
-# ::
-# 
-#    saved_model = GarmentClassifier()
-#    saved_model.load_state_dict(torch.load(PATH))
-# 
+#
+# .. code:: python
+#
+#     saved_model = GarmentClassifier()
+#     saved_model.load_state_dict(torch.load(PATH))
+#
 # Once you’ve loaded the model, it’s ready for whatever you need it for -
 # more training, inference, or analysis.
 #