From 5984d3f0b4586e7b3b31ecbef95ed998c6637b81 Mon Sep 17 00:00:00 2001
From: SandraGH5 <80421934+SandraGH5@users.noreply.github.com>
Date: Tue, 21 Sep 2021 12:52:43 -0700
Subject: [PATCH] Update pytorch_single_node_and_gpu.py (#419)

Embedding a video to outline the Pytorch training process for single-node single GPUs.
---
 .../mnist_classifier/pytorch_single_node_and_gpu.py           | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/cookbook/case_studies/ml_training/mnist_classifier/pytorch_single_node_and_gpu.py b/cookbook/case_studies/ml_training/mnist_classifier/pytorch_single_node_and_gpu.py
index 8c61a4b0e0..d294fbcf8d 100644
--- a/cookbook/case_studies/ml_training/mnist_classifier/pytorch_single_node_and_gpu.py
+++ b/cookbook/case_studies/ml_training/mnist_classifier/pytorch_single_node_and_gpu.py
@@ -10,7 +10,9 @@
 that supports *PyTorch-Ignite* or *PyTorch Lightening* support, but this is not critical). One can request for a GPU, simply
 by setting GPU="1" resource request and then at runtime, the GPU will be provisioned.
 
-In this example, we'll see how we can create any PyTorch model, train it using Flyte and a specialized container. 
+In this example, we'll see how we can create any PyTorch model, train it using Flyte and a specialized container. The following video will outline the basics of this process.
+
+..  youtube:: sJrERMVtxL4
 """
 
 # %%