Update src/sagemaker/huggingface/estimator.py

Lokiiiiii · mchoi8739 · web-flow · commit 2ea39c44551b · 2022-08-22T16:52:30.000-07:00
Co-authored-by: Miyoung &lt;myoung8739@gmail.com&gt;
diff --git a/src/sagemaker/huggingface/estimator.py b/src/sagemaker/huggingface/estimator.py
@@ -151,6 +151,12 @@ def __init__(
                             "enabled": True
                         }
                     }
+                    
+                To learn more, see `SageMaker Training Compiler <https://docs.aws.amazon.com/sagemaker/latest/dg/training-compiler.html>`_ in the *Amazon SageMaker Developer Guide*.
+                
+                .. note::
+                
+                    When you use this PyTorch XLA option for distributed training strategy, you must add the ``compiler_config`` parameter and activate SageMaker Training Compiler.
             compiler_config (:class:`~sagemaker.huggingface.TrainingCompilerConfig`):
                 Configures SageMaker Training Compiler to accelerate training.
 

Original file line number	Diff line number	Diff line change
`@@ -151,6 +151,12 @@ def __init__(`
`151`	`151`	`"enabled": True`
`152`	`152`	`}`
`153`	`153`	`}`
	`154`	`+`
	`155`	+ To learn more, see `SageMaker Training Compiler <https://docs.aws.amazon.com/sagemaker/latest/dg/training-compiler.html>`_ in the Amazon SageMaker Developer Guide.
	`156`	`+`
	`157`	`+ .. note::`
	`158`	`+`
	`159`	+ When you use this PyTorch XLA option for distributed training strategy, you must add the ``compiler_config`` parameter and activate SageMaker Training Compiler.
`154`	`160`	compiler_config (:class:`~sagemaker.huggingface.TrainingCompilerConfig`):
`155`	`161`	`Configures SageMaker Training Compiler to accelerate training.`
`156`	`162`