add a note

mchoi8739 · navinsoni · commit 5a10b34b29c8 · 2023-03-08T09:34:03.000-08:00
diff --git a/doc/frameworks/pytorch/using_pytorch.rst b/doc/frameworks/pytorch/using_pytorch.rst
@@ -315,6 +315,13 @@ to run a distributed training job on two ``ml.p4d.24xlarge`` instances.
         }
     )
 
+
+.. note:: 
+    
+    For more information about setting up ``torchrun`` in your training script,
+    see `torchrun (Elastic Launch) <https://pytorch.org/docs/stable/elastic/run.html>_` in *the
+    PyTorch documentation*.
+
 ----
 
 .. _distributed-pytorch-training-on-trainium:

Original file line number	Diff line number	Diff line change
@@ -315,6 +315,13 @@ to run a distributed training job on two ``ml.p4d.24xlarge`` instances.
`315`	`315`	`}`
`316`	`316`	`)`
`317`	`317`
	`318`	`+`
	`319`	`+.. note::`
	`320`	`+`
	`321`	+ For more information about setting up ``torchrun`` in your training script,
	`322`	+ see `torchrun (Elastic Launch) <https://pytorch.org/docs/stable/elastic/run.html>_` in *the
	`323`	`+ PyTorch documentation*.`
	`324`	`+`
`318`	`325`	`----`
`319`	`326`
`320`	`327`	`.. _distributed-pytorch-training-on-trainium:`