diff --git a/src/sagemaker/debugger/framework_profile.py b/src/sagemaker/debugger/framework_profile.py index cae16544af..0842b5e63a 100644 --- a/src/sagemaker/debugger/framework_profile.py +++ b/src/sagemaker/debugger/framework_profile.py @@ -143,6 +143,12 @@ def __init__( profiling. Configure it using the :class:`~sagemaker.debugger.metrics_config.DetailedProfilingConfig` class. Pass ``DetailedProfilingConfig()`` to use the default configuration. + + .. warning:: + This detailed framework profiling feature discontinues support for TensorFlow v2.11 + and later. To use the detailed profiling feature, use previous versions of + TensorFlow between v2.3.1 and v2.10.0. + dataloader_profiling_config (DataloaderProfilingConfig): The configuration for dataloader metrics profiling. Configure it using the :class:`~sagemaker.debugger.metrics_config.DataloaderProfilingConfig` class. diff --git a/src/sagemaker/debugger/metrics_config.py b/src/sagemaker/debugger/metrics_config.py index 5b26d5c8eb..c3aa3793dd 100644 --- a/src/sagemaker/debugger/metrics_config.py +++ b/src/sagemaker/debugger/metrics_config.py @@ -203,8 +203,7 @@ def __init__( ): """Specify target steps or a target duration to profile. - By default, it profiles step 5 - of training. + By default, it profiles step 5 of the training job. If **profile_default_steps** is set to `True` and none of the other range parameters is specified, @@ -224,6 +223,11 @@ def __init__( if one of the two pairs is used. If both pairs are specified, a conflict error occurs. + .. warning:: + This detailed framework profiling feature discontinues support for TensorFlow v2.11 + and later. To use the detailed profiling feature, use previous versions of + TensorFlow between v2.3.1 and v2.10.0. + """ assert isinstance( profile_default_steps, bool