From 2bd5623e7e004f9c130ff1f5c20633319243d288 Mon Sep 17 00:00:00 2001 From: Jonathan Makunga Date: Wed, 29 May 2024 09:49:04 -0700 Subject: [PATCH 01/11] Model data arn --- src/sagemaker/jumpstart/types.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/sagemaker/jumpstart/types.py b/src/sagemaker/jumpstart/types.py index 7a43f0ff67..dedc7ceabb 100644 --- a/src/sagemaker/jumpstart/types.py +++ b/src/sagemaker/jumpstart/types.py @@ -2284,6 +2284,7 @@ class DeploymentArgs(BaseDeploymentConfigDataHolder): __slots__ = [ "image_uri", "model_data", + "model_package_arn", "environment", "instance_type", "compute_resource_requirements", @@ -2301,6 +2302,7 @@ def __init__( if init_kwargs is not None: self.image_uri = init_kwargs.image_uri self.model_data = init_kwargs.model_data + self.model_package_arn = init_kwargs.model_package_arn self.instance_type = init_kwargs.instance_type self.environment = init_kwargs.env if init_kwargs.resources is not None: From 2bc731d73336d55f26381a8417ff2c0021a21412 Mon Sep 17 00:00:00 2001 From: Jonathan Makunga Date: Wed, 29 May 2024 10:18:48 -0700 Subject: [PATCH 02/11] Refactoring --- src/sagemaker/jumpstart/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/sagemaker/jumpstart/utils.py b/src/sagemaker/jumpstart/utils.py index 7de79204aa..a1b5d7fa9a 100644 --- a/src/sagemaker/jumpstart/utils.py +++ b/src/sagemaker/jumpstart/utils.py @@ -1284,7 +1284,7 @@ def wrapped_f(*args, **kwargs): break elif isinstance(res, dict): keys = list(res.keys()) - if "Instance Rate" not in keys[-1]: + if len(keys) == 0 or "Instance Rate" not in keys[-1]: f.cache_clear() elif len(res[keys[1]]) > len(res[keys[-1]]): del res[keys[-1]] From bc057212d411757d90776f6ff345dfecd7ace396 Mon Sep 17 00:00:00 2001 From: Jonathan Makunga Date: Wed, 29 May 2024 13:59:10 -0700 Subject: [PATCH 03/11] Refactoring --- src/sagemaker/jumpstart/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/sagemaker/jumpstart/utils.py b/src/sagemaker/jumpstart/utils.py index a1b5d7fa9a..cf40d042b5 100644 --- a/src/sagemaker/jumpstart/utils.py +++ b/src/sagemaker/jumpstart/utils.py @@ -1129,7 +1129,7 @@ def get_metrics_from_deployment_configs( if not deployment_configs: return {} - data = {"Instance Type": [], "Config Name": [], "Concurrent Users": []} + data = {"Instance Type": [], "Concurrent Users": [], "Config Name": []} instance_rate_data = {} for index, deployment_config in enumerate(deployment_configs): benchmark_metrics = deployment_config.benchmark_metrics From 9462dc9709b5388a7ca85e1a716ef8139aa1eaac Mon Sep 17 00:00:00 2001 From: Jonathan Makunga Date: Wed, 29 May 2024 14:16:30 -0700 Subject: [PATCH 04/11] acceleration_configs --- src/sagemaker/jumpstart/model.py | 6 ++---- src/sagemaker/jumpstart/types.py | 12 ++++++------ 2 files changed, 8 insertions(+), 10 deletions(-) diff --git a/src/sagemaker/jumpstart/model.py b/src/sagemaker/jumpstart/model.py index 534e93c285..6bdd198513 100644 --- a/src/sagemaker/jumpstart/model.py +++ b/src/sagemaker/jumpstart/model.py @@ -898,11 +898,10 @@ def _get_deployment_configs( err = None for config_name, metadata_config in self._metadata_configs.items(): - resolved_config = metadata_config.resolved_config if selected_config_name == config_name: instance_type_to_use = selected_instance_type else: - instance_type_to_use = resolved_config.get("default_inference_instance_type") + instance_type_to_use = metadata_config.resolved_config.get("default_inference_instance_type") if metadata_config.benchmark_metrics: err, metadata_config.benchmark_metrics = ( @@ -941,8 +940,7 @@ def _get_deployment_configs( deployment_config_metadata = DeploymentConfigMetadata( config_name, - metadata_config.benchmark_metrics, - resolved_config, + metadata_config, init_kwargs, deploy_kwargs, ) diff --git a/src/sagemaker/jumpstart/types.py b/src/sagemaker/jumpstart/types.py index dedc7ceabb..8be38b89cc 100644 --- a/src/sagemaker/jumpstart/types.py +++ b/src/sagemaker/jumpstart/types.py @@ -1075,6 +1075,7 @@ class JumpStartMetadataConfig(JumpStartDataHolderType): __slots__ = [ "base_fields", "benchmark_metrics", + "acceleration_configs", "config_components", "resolved_metadata_config", "config_name", @@ -1112,6 +1113,7 @@ def __init__( if config and config.get("benchmark_metrics") else None ) + self.acceleration_configs = config.get("acceleration_configs") self.resolved_metadata_config: Optional[Dict[str, Any]] = None self.config_name: Optional[str] = config_name self.default_inference_config: Optional[str] = config.get("default_inference_config") @@ -2334,14 +2336,12 @@ class DeploymentConfigMetadata(BaseDeploymentConfigDataHolder): def __init__( self, config_name: Optional[str] = None, - benchmark_metrics: Optional[Dict[str, List[JumpStartBenchmarkStat]]] = None, - resolved_config: Optional[Dict[str, Any]] = None, + metadata_config: Optional[JumpStartMetadataConfig] = None, init_kwargs: Optional[JumpStartModelInitKwargs] = None, deploy_kwargs: Optional[JumpStartModelDeployKwargs] = None, ): """Instantiates DeploymentConfigMetadata object.""" self.deployment_config_name = config_name - self.deployment_args = DeploymentArgs(init_kwargs, deploy_kwargs, resolved_config) - self.benchmark_metrics = benchmark_metrics - if resolved_config is not None: - self.acceleration_configs = resolved_config.get("acceleration_configs") + self.deployment_args = DeploymentArgs(init_kwargs, deploy_kwargs, metadata_config.resolved_config) + self.benchmark_metrics = metadata_config.benchmark_metrics + self.acceleration_configs = metadata_config.acceleration_configs From 06b02c8827641ca4365dfebf251da8da68fc27af Mon Sep 17 00:00:00 2001 From: Jonathan Makunga Date: Wed, 29 May 2024 15:38:15 -0700 Subject: [PATCH 05/11] Refactoring --- src/sagemaker/jumpstart/model.py | 4 +++- src/sagemaker/jumpstart/types.py | 4 +++- src/sagemaker/jumpstart/utils.py | 2 +- 3 files changed, 7 insertions(+), 3 deletions(-) diff --git a/src/sagemaker/jumpstart/model.py b/src/sagemaker/jumpstart/model.py index 6bdd198513..96bdd99e5f 100644 --- a/src/sagemaker/jumpstart/model.py +++ b/src/sagemaker/jumpstart/model.py @@ -901,7 +901,9 @@ def _get_deployment_configs( if selected_config_name == config_name: instance_type_to_use = selected_instance_type else: - instance_type_to_use = metadata_config.resolved_config.get("default_inference_instance_type") + instance_type_to_use = metadata_config.resolved_config.get( + "default_inference_instance_type" + ) if metadata_config.benchmark_metrics: err, metadata_config.benchmark_metrics = ( diff --git a/src/sagemaker/jumpstart/types.py b/src/sagemaker/jumpstart/types.py index 26b52f74f3..04e8b91e26 100644 --- a/src/sagemaker/jumpstart/types.py +++ b/src/sagemaker/jumpstart/types.py @@ -2351,6 +2351,8 @@ def __init__( ): """Instantiates DeploymentConfigMetadata object.""" self.deployment_config_name = config_name - self.deployment_args = DeploymentArgs(init_kwargs, deploy_kwargs, metadata_config.resolved_config) + self.deployment_args = DeploymentArgs( + init_kwargs, deploy_kwargs, metadata_config.resolved_config + ) self.benchmark_metrics = metadata_config.benchmark_metrics self.acceleration_configs = metadata_config.acceleration_configs diff --git a/src/sagemaker/jumpstart/utils.py b/src/sagemaker/jumpstart/utils.py index cf40d042b5..a1b5d7fa9a 100644 --- a/src/sagemaker/jumpstart/utils.py +++ b/src/sagemaker/jumpstart/utils.py @@ -1129,7 +1129,7 @@ def get_metrics_from_deployment_configs( if not deployment_configs: return {} - data = {"Instance Type": [], "Concurrent Users": [], "Config Name": []} + data = {"Instance Type": [], "Config Name": [], "Concurrent Users": []} instance_rate_data = {} for index, deployment_config in enumerate(deployment_configs): benchmark_metrics = deployment_config.benchmark_metrics From 28ceac2fbd5da632ea82380f386d676581498f2f Mon Sep 17 00:00:00 2001 From: Jonathan Makunga Date: Thu, 30 May 2024 09:18:23 -0700 Subject: [PATCH 06/11] UT --- tests/unit/sagemaker/jumpstart/test_types.py | 3 +-- tests/unit/sagemaker/jumpstart/utils.py | 3 +-- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/unit/sagemaker/jumpstart/test_types.py b/tests/unit/sagemaker/jumpstart/test_types.py index 7f2c7b2aad..23fa42c09a 100644 --- a/tests/unit/sagemaker/jumpstart/test_types.py +++ b/tests/unit/sagemaker/jumpstart/test_types.py @@ -1372,8 +1372,7 @@ def test_deployment_config_metadata(): deployment_config_metadata = DeploymentConfigMetadata( jumpstart_config.config_name, - jumpstart_config.benchmark_metrics, - jumpstart_config.resolved_config, + jumpstart_config, JumpStartModelInitKwargs( model_id=specs.model_id, model_data=INIT_KWARGS.get("model_data"), diff --git a/tests/unit/sagemaker/jumpstart/utils.py b/tests/unit/sagemaker/jumpstart/utils.py index 97ee36e998..cc4ef71cee 100644 --- a/tests/unit/sagemaker/jumpstart/utils.py +++ b/tests/unit/sagemaker/jumpstart/utils.py @@ -378,8 +378,7 @@ def get_base_deployment_configs_metadata( configs.append( DeploymentConfigMetadata( config_name=config_name, - benchmark_metrics=jumpstart_config.benchmark_metrics, - resolved_config=jumpstart_config.resolved_config, + metadata_config=jumpstart_config, init_kwargs=get_mock_init_kwargs( get_base_spec_with_prototype_configs().model_id, config_name ), From 19343dcf297303d9508686fd620e44e43c738ad5 Mon Sep 17 00:00:00 2001 From: Jonathan Makunga Date: Thu, 30 May 2024 13:15:53 -0700 Subject: [PATCH 07/11] Add Filter --- src/sagemaker/jumpstart/model.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/src/sagemaker/jumpstart/model.py b/src/sagemaker/jumpstart/model.py index 96bdd99e5f..81efc1f17a 100644 --- a/src/sagemaker/jumpstart/model.py +++ b/src/sagemaker/jumpstart/model.py @@ -469,9 +469,15 @@ def benchmark_metrics(self) -> pd.DataFrame: df.index = blank_index return df - def display_benchmark_metrics(self, *args, **kwargs) -> None: + def display_benchmark_metrics(self, **kwargs) -> None: """Display deployment configs benchmark metrics.""" - print(self.benchmark_metrics.to_markdown(index=False, floatfmt=".2f"), *args, **kwargs) + df = self.benchmark_metrics + + instance_type = kwargs.get("instance_type") + if instance_type: + df = df[df["Instance Type"].str.contains(instance_type)] + + print(df.to_markdown(index=False, floatfmt=".2f")) def list_deployment_configs(self) -> List[Dict[str, Any]]: """List deployment configs for ``This`` model. From 283c4a8a63b7fc5967fcc9bc99f729cde8f24196 Mon Sep 17 00:00:00 2001 From: Jonathan Makunga Date: Thu, 30 May 2024 13:46:55 -0700 Subject: [PATCH 08/11] UT --- .../text/machine_translation_hugging_face.rst | 41 ++ .../text/question_answering_pytorch.rst | 101 +++ ...tence_pair_classification_hugging_face.rst | 116 +++ ...entence_pair_classification_tensorflow.rst | 61 ++ .../text/text_classification_tensorflow.rst | 21 + .../text/text_generation_hugging_face.rst | 96 +++ .../text/text_summarization_hugging_face.rst | 51 ++ .../vision/image_classification_pytorch.rst | 146 ++++ .../image_classification_tensorflow.rst | 686 ++++++++++++++++++ .../vision/image_embedding_tensorflow.rst | 271 +++++++ .../vision/instance_segmentation_mxnet.rst | 31 + .../vision/object_detection_mxnet.rst | 96 +++ .../vision/object_detection_pytorch.rst | 16 + .../vision/object_detection_tensorflow.rst | 191 +++++ .../vision/semantic_segmentation_mxnet.rst | 31 + .../text_embedding_tensorflow_mxnet.rst | 31 + .../sagemaker/jumpstart/model/test_model.py | 2 + 17 files changed, 1988 insertions(+) diff --git a/doc/algorithms/text/machine_translation_hugging_face.rst b/doc/algorithms/text/machine_translation_hugging_face.rst index d533d0e64d..e6a8ab6893 100644 --- a/doc/algorithms/text/machine_translation_hugging_face.rst +++ b/doc/algorithms/text/machine_translation_hugging_face.rst @@ -8,3 +8,44 @@ This is a supervised machine translation algorithm which supports many pre-train demonstrates how to use the Sagemaker Python SDK for Machine Translation for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK `. + +.. list-table:: Available Models + :widths: 50 20 20 20 20 + :header-rows: 1 + :class: datatable + + * - Model ID + - Fine Tunable? + - Latest Version + - Min SDK Version + - Source + * - huggingface-translation-opus-mt-en-es + - False + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-translation-opus-mt-en-vi + - False + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-translation-opus-mt-mul-en + - False + - 1.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-translation-t5-base + - False + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-translation-t5-large + - False + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-translation-t5-small + - False + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ diff --git a/doc/algorithms/text/question_answering_pytorch.rst b/doc/algorithms/text/question_answering_pytorch.rst index 9d9d74ccb1..4ad2205f40 100644 --- a/doc/algorithms/text/question_answering_pytorch.rst +++ b/doc/algorithms/text/question_answering_pytorch.rst @@ -7,3 +7,104 @@ This is a supervised question answering algorithm which supports fine-tuning of demonstrates how to use the Sagemaker Python SDK for Question Answering for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` + +.. list-table:: Available Models + :widths: 50 20 20 20 20 + :header-rows: 1 + :class: datatable + + * - Model ID + - Fine Tunable? + - Latest Version + - Min SDK Version + - Source + * - pytorch-eqa-bert-base-cased + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ + * - pytorch-eqa-bert-base-multilingual-cased + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ + * - pytorch-eqa-bert-base-multilingual-uncased + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ + * - pytorch-eqa-bert-base-uncased + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ + * - pytorch-eqa-bert-large-cased + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ + * - pytorch-eqa-bert-large-cased-whole-word-masking + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ + * - pytorch-eqa-bert-large-cased-whole-word-masking-finetuned-squad + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ + * - pytorch-eqa-bert-large-uncased + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ + * - pytorch-eqa-bert-large-uncased-whole-word-masking + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ + * - pytorch-eqa-bert-large-uncased-whole-word-masking-finetuned-squad + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ + * - pytorch-eqa-distilbert-base-cased + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ + * - pytorch-eqa-distilbert-base-multilingual-cased + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ + * - pytorch-eqa-distilbert-base-uncased + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ + * - pytorch-eqa-distilroberta-base + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ + * - pytorch-eqa-roberta-base + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ + * - pytorch-eqa-roberta-base-openai-detector + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ + * - pytorch-eqa-roberta-large + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ + * - pytorch-eqa-roberta-large-openai-detector + - True + - 1.2.1 + - 2.75.0 + - `Pytorch Hub `__ diff --git a/doc/algorithms/text/sentence_pair_classification_hugging_face.rst b/doc/algorithms/text/sentence_pair_classification_hugging_face.rst index 2892b9d516..a96bb9fac0 100644 --- a/doc/algorithms/text/sentence_pair_classification_hugging_face.rst +++ b/doc/algorithms/text/sentence_pair_classification_hugging_face.rst @@ -7,3 +7,119 @@ This is a supervised sentence pair classification algorithm which supports fine- demonstrates how to use the Sagemaker Python SDK for Sentence Pair Classification for using these algorithms. For detailed documentation please refer `Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK `__ + +.. list-table:: Available Models + :widths: 50 20 20 20 20 + :header-rows: 1 + :class: datatable + + * - Model ID + - Fine Tunable? + - Latest Version + - Min SDK Version + - Source + * - huggingface-spc-bert-base-cased + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-bert-base-multilingual-cased + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-bert-base-multilingual-uncased + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-bert-base-uncased + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-bert-large-cased + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-bert-large-cased-whole-word-masking + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-bert-large-uncased + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-bert-large-uncased-whole-word-masking + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-distilbert-base-cased + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-distilbert-base-multilingual-cased + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-distilbert-base-uncased + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-distilroberta-base + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-roberta-base + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-roberta-base-openai-detector + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-roberta-large + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-roberta-large-openai-detector + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-xlm-clm-ende-1024 + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-xlm-mlm-ende-1024 + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-xlm-mlm-enro-1024 + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-xlm-mlm-tlm-xnli15-1024 + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-spc-xlm-mlm-xnli15-1024 + - True + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ diff --git a/doc/algorithms/text/sentence_pair_classification_tensorflow.rst b/doc/algorithms/text/sentence_pair_classification_tensorflow.rst index 80264e84f3..7b4a079dd5 100644 --- a/doc/algorithms/text/sentence_pair_classification_tensorflow.rst +++ b/doc/algorithms/text/sentence_pair_classification_tensorflow.rst @@ -7,3 +7,64 @@ This is a supervised sentence pair classification algorithm which supports fine- demonstrates how to use the Sagemaker Python SDK for Sentence Pair Classification for using these algorithms. For detailed documentation please refer `Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK `__ + +.. list-table:: Available Models + :widths: 50 20 20 20 20 + :header-rows: 1 + :class: datatable + + * - Model ID + - Fine Tunable? + - Latest Version + - Min SDK Version + - Source + * - tensorflow-spc-bert-en-cased-L-12-H-768-A-12-2 + - True + - 2.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-spc-bert-en-uncased-L-12-H-768-A-12-2 + - True + - 2.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-spc-bert-en-uncased-L-24-H-1024-A-16-2 + - True + - 2.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-spc-bert-en-wwm-cased-L-24-H-1024-A-16-2 + - True + - 2.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-spc-bert-en-wwm-uncased-L-24-H-1024-A-16-2 + - True + - 2.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-spc-bert-multi-cased-L-12-H-768-A-12-2 + - True + - 2.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-spc-electra-base-1 + - True + - 2.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-spc-electra-small-1 + - True + - 2.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-spc-experts-bert-pubmed-1 + - True + - 2.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-spc-experts-bert-wiki-books-1 + - True + - 2.0.0 + - 2.189.0 + - `Tensorflow Hub `__ diff --git a/doc/algorithms/text/text_classification_tensorflow.rst b/doc/algorithms/text/text_classification_tensorflow.rst index c60a5b3e1c..3536556bb8 100644 --- a/doc/algorithms/text/text_classification_tensorflow.rst +++ b/doc/algorithms/text/text_classification_tensorflow.rst @@ -7,3 +7,24 @@ This is a supervised text classification algorithm which supports fine-tuning of demonstrates how to use the Sagemaker Python SDK for Text Classification for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` + +.. list-table:: Available Models + :widths: 50 20 20 20 20 + :header-rows: 1 + :class: datatable + + * - Model ID + - Fine Tunable? + - Latest Version + - Min SDK Version + - Source + * - tensorflow-tc-albert-en-base + - True + - 2.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-tc-bert-en-cased-L-12-H-768-A-12-2 + - True + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ diff --git a/doc/algorithms/text/text_generation_hugging_face.rst b/doc/algorithms/text/text_generation_hugging_face.rst index 30fae26196..a865f75c65 100644 --- a/doc/algorithms/text/text_generation_hugging_face.rst +++ b/doc/algorithms/text/text_generation_hugging_face.rst @@ -7,3 +7,99 @@ This is a supervised text generation algorithm which supports many pre-trained m demonstrates how to use the Sagemaker Python SDK for Text Generation for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` + +.. list-table:: Available Models + :widths: 50 20 20 20 20 + :header-rows: 1 + :class: datatable + + * - Model ID + - Fine Tunable? + - Latest Version + - Min SDK Version + - Source + * - huggingface-textgeneration-bloom-1b1 + - False + - 2.1.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-textgeneration-bloom-1b7 + - False + - 2.1.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-textgeneration-bloom-560m + - False + - 2.1.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-textgeneration-bloomz-1b1 + - False + - 2.1.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-textgeneration-bloomz-1b7 + - False + - 2.1.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-textgeneration-bloomz-560m + - False + - 2.1.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-textgeneration-distilgpt2 + - False + - 2.1.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-textgeneration-dolly-v2-12b-bf16 + - False + - 2.1.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-textgeneration-dolly-v2-3b-bf16 + - False + - 2.1.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-textgeneration-dolly-v2-7b-bf16 + - False + - 2.1.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-textgeneration-falcon-40b-bf16 + - False + - 1.0.0 + - 2.144.0 + - `HuggingFace `__ + * - huggingface-textgeneration-falcon-40b-instruct-bf16 + - False + - 1.0.0 + - 2.144.0 + - `HuggingFace `__ + * - huggingface-textgeneration-falcon-7b-bf16 + - False + - 1.0.0 + - 2.144.0 + - `HuggingFace `__ + * - huggingface-textgeneration-falcon-7b-instruct-bf16 + - False + - 1.0.0 + - 2.144.0 + - `HuggingFace `__ + * - huggingface-textgeneration-gpt2 + - False + - 2.1.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-textgeneration-models + - False + - 1.3.0 + - 2.144.0 + - `HuggingFace `__ + * - huggingface-textgeneration-open-llama + - False + - 3.1.0 + - 2.189.0 + - `HuggingFace `__ diff --git a/doc/algorithms/text/text_summarization_hugging_face.rst b/doc/algorithms/text/text_summarization_hugging_face.rst index 206c880ba3..af0346b146 100644 --- a/doc/algorithms/text/text_summarization_hugging_face.rst +++ b/doc/algorithms/text/text_summarization_hugging_face.rst @@ -7,3 +7,54 @@ This is a supervised text summarization algorithm which supports many pre-traine demonstrates how to use the Sagemaker Python SDK for Text Summarization for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` + +.. list-table:: Available Models + :widths: 50 20 20 20 20 + :header-rows: 1 + :class: datatable + + * - Model ID + - Fine Tunable? + - Latest Version + - Min SDK Version + - Source + * - huggingface-summarization-bart-large-cnn-samsum + - False + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-summarization-bert-small2bert-small-finetuned-cnn-daily-mail-summarization + - False + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-summarization-bigbird-pegasus-large-arxiv + - False + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-summarization-bigbird-pegasus-large-pubmed + - False + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-summarization-distilbart-cnn-12-6 + - False + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-summarization-distilbart-cnn-6-6 + - False + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-summarization-distilbart-xsum-1-1 + - False + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ + * - huggingface-summarization-distilbart-xsum-12-3 + - False + - 2.0.0 + - 2.189.0 + - `HuggingFace `__ diff --git a/doc/algorithms/vision/image_classification_pytorch.rst b/doc/algorithms/vision/image_classification_pytorch.rst index 3c154c6cfe..88a8e67571 100644 --- a/doc/algorithms/vision/image_classification_pytorch.rst +++ b/doc/algorithms/vision/image_classification_pytorch.rst @@ -7,3 +7,149 @@ This is a supervised image clasification algorithm which supports fine-tuning of demonstrates how to use the Sagemaker Python SDK for Image Classification for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` + +.. list-table:: Available Models + :widths: 50 20 20 20 20 + :header-rows: 1 + :class: datatable + + * - Model ID + - Fine Tunable? + - Latest Version + - Min SDK Version + - Source + * - pytorch-ic-alexnet + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-densenet121 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-densenet161 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-densenet169 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-densenet201 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-googlenet + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-mobilenet-v2 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-resnet101 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-resnet152 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-resnet18 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-resnet34 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-resnet50 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-resnext101-32x8d + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-resnext50-32x4d + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-shufflenet-v2-x1-0 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-squeezenet1-0 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-squeezenet1-1 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-vgg11 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-vgg11-bn + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-vgg13 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-vgg13-bn + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-vgg16 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-vgg16-bn + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-vgg19 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-vgg19-bn + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-wide-resnet101-2 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ + * - pytorch-ic-wide-resnet50-2 + - True + - 3.0.0 + - 2.189.0 + - `Pytorch Hub `__ diff --git a/doc/algorithms/vision/image_classification_tensorflow.rst b/doc/algorithms/vision/image_classification_tensorflow.rst index e49820ee50..5f6cd309be 100644 --- a/doc/algorithms/vision/image_classification_tensorflow.rst +++ b/doc/algorithms/vision/image_classification_tensorflow.rst @@ -7,3 +7,689 @@ This is a supervised image clasification algorithm which supports fine-tuning of demonstrates how to use the Sagemaker Python SDK for Image Classification for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` + +.. list-table:: Available Models + :widths: 50 20 20 20 20 + :header-rows: 1 + :class: datatable + + * - Model ID + - Fine Tunable? + - Latest Version + - Min SDK Version + - Source + * - tensorflow-ic-bit-m-r101x1-ilsvrc2012-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-bit-m-r101x1-imagenet21k-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-bit-m-r101x3-ilsvrc2012-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-bit-m-r101x3-imagenet21k-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-bit-m-r152x4-ilsvrc2012 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-bit-m-r152x4-imagenet21k + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-bit-m-r50x1-ilsvrc2012-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-bit-m-r50x1-imagenet21k-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-bit-m-r50x3-ilsvrc2012-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-bit-m-r50x3-imagenet21k-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-bit-s-r101x1-ilsvrc2012-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-bit-s-r101x3-ilsvrc2012-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-bit-s-r152x4-ilsvrc2012 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-bit-s-r50x1-ilsvrc2012-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-bit-s-r50x3-ilsvrc2012-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-cait-m36-384 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-cait-m48-448 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-cait-s24-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-cait-s24-384 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-cait-s36-384 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-cait-xs24-384 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-cait-xxs24-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-cait-xxs24-384 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-cait-xxs36-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-cait-xxs36-384 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-deit-base-distilled-patch16-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-deit-base-distilled-patch16-384 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-deit-base-patch16-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-deit-base-patch16-384 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-deit-small-distilled-patch16-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-deit-small-patch16-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-deit-tiny-distilled-patch16-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-deit-tiny-patch16-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-b0-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-b1-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-b2-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-b3-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-b4-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-b5-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-b6-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-b7-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-lite0-classification-2 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-lite1-classification-2 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-lite2-classification-2 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-lite3-classification-2 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-lite4-classification-2 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet1k-b0 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet1k-b1 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet1k-b2 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet1k-b3 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet1k-l + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet1k-m + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet1k-s + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet21k-b0 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet21k-b1 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet21k-b2 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet21k-b3 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-b0 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-b1 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-b2 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-b3 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-l + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-m + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-s + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-xl + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet21k-l + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet21k-m + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet21k-s + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-efficientnet-v2-imagenet21k-xl + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-inception-resnet-v2-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-inception-v1-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-inception-v2-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-inception-v3-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v1-025-128-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v1-025-160-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v1-025-192-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v1-025-224-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v1-050-128-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v1-050-160-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v1-050-192-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v1-050-224-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v1-075-128-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v1-075-160-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v1-075-192-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v1-075-224-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v1-100-128-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v1-100-160-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v1-100-192-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v1-100-224-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-035-128 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-035-160 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-035-192 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-035-224-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-035-96 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-050-128 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-050-160 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-050-192 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-050-224-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-050-96 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-075-128 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-075-160 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-075-192 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-075-224-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-075-96 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-100-160 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-100-192 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-100-224-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-100-96 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-130-224-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v2-140-224-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v3-large-075-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v3-large-100-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v3-small-075-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-mobilenet-v3-small-100-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-nasnet-large + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-nasnet-mobile + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-pnasnet-large + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-resnet-v1-101-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-resnet-v1-152-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-resnet-v1-50-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-resnet-v2-101-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-resnet-v2-152-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-imagenet-resnet-v2-50-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-resnet-50-classification-1 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-swin-base-patch4-window12-384 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-swin-base-patch4-window7-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-swin-large-patch4-window12-384 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-swin-large-patch4-window7-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-swin-s3-base-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-swin-s3-small-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-swin-s3-tiny-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-swin-small-patch4-window7-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-swin-tiny-patch4-window7-224 + - True + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-tf2-preview-inception-v3-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-ic-tf2-preview-mobilenet-v2-classification-4 + - True + - 4.0.2 + - 2.189.0 + - `Tensorflow Hub `__ diff --git a/doc/algorithms/vision/image_embedding_tensorflow.rst b/doc/algorithms/vision/image_embedding_tensorflow.rst index 0938377354..bfba13686c 100644 --- a/doc/algorithms/vision/image_embedding_tensorflow.rst +++ b/doc/algorithms/vision/image_embedding_tensorflow.rst @@ -7,3 +7,274 @@ This is a supervised image embedding algorithm which supports many pre-trained m demonstrates how to use the Sagemaker Python SDK for Image Embedding for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` + +.. list-table:: Available Models + :widths: 50 20 20 20 20 + :header-rows: 1 + :class: datatable + + * - Model ID + - Fine Tunable? + - Latest Version + - Min SDK Version + - Source + * - tensorflow-icembedding-bit-m-r101x1-ilsvrc2012-featurevector-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-bit-m-r101x3-imagenet21k-featurevector-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-bit-m-r50x1-ilsvrc2012-featurevector-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-bit-m-r50x3-imagenet21k-featurevector-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-bit-s-r101x1-ilsvrc2012-featurevector-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-bit-s-r101x3-ilsvrc2012-featurevector-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-bit-s-r50x1-ilsvrc2012-featurevector-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-bit-s-r50x3-ilsvrc2012-featurevector-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-efficientnet-b0-featurevector-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-efficientnet-b1-featurevector-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-efficientnet-b2-featurevector-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-efficientnet-b3-featurevector-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-efficientnet-b6-featurevector-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-efficientnet-lite0-featurevector-2 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-efficientnet-lite1-featurevector-2 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-efficientnet-lite2-featurevector-2 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-efficientnet-lite3-featurevector-2 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-efficientnet-lite4-featurevector-2 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-inception-v1-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-inception-v2-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-inception-v3-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v1-025-128-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v1-025-160-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v1-025-192-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v1-025-224-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v1-050-128-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v1-050-160-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v1-050-192-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v1-050-224-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v1-075-128-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v1-075-160-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v1-075-192-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v1-075-224-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v1-100-128-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v1-100-160-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v1-100-192-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v1-100-224-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v2-035-224-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v2-050-224-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v2-075-224-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v2-100-224-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v2-130-224-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-mobilenet-v2-140-224-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-resnet-v1-101-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-resnet-v1-152-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-resnet-v1-50-featurevector-4 + - False + - 2.0.2 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-resnet-v2-101-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-resnet-v2-152-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-imagenet-resnet-v2-50-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-resnet-50-featurevector-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-tf2-preview-inception-v3-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-icembedding-tf2-preview-mobilenet-v2-featurevector-4 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ diff --git a/doc/algorithms/vision/instance_segmentation_mxnet.rst b/doc/algorithms/vision/instance_segmentation_mxnet.rst index a38611bc9a..33aa87fe28 100644 --- a/doc/algorithms/vision/instance_segmentation_mxnet.rst +++ b/doc/algorithms/vision/instance_segmentation_mxnet.rst @@ -7,3 +7,34 @@ This is a supervised image segmentation algorithm which supports many pre-train demonstrates how to use the Sagemaker Python SDK for Image Segmentation for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` + +.. list-table:: Available Models + :widths: 50 20 20 20 20 + :header-rows: 1 + :class: datatable + + * - Model ID + - Fine Tunable? + - Latest Version + - Min SDK Version + - Source + * - mxnet-is-mask-rcnn-fpn-resnet101-v1d-coco + - False + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-is-mask-rcnn-fpn-resnet18-v1b-coco + - False + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-is-mask-rcnn-fpn-resnet50-v1b-coco + - False + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-is-mask-rcnn-resnet18-v1b-coco + - False + - 2.0.0 + - 2.189.0 + - `GluonCV `__ diff --git a/doc/algorithms/vision/object_detection_mxnet.rst b/doc/algorithms/vision/object_detection_mxnet.rst index 9ce52f992b..70d6b55178 100644 --- a/doc/algorithms/vision/object_detection_mxnet.rst +++ b/doc/algorithms/vision/object_detection_mxnet.rst @@ -7,3 +7,99 @@ This is a supervised object detection algorithm which supports fine-tuning of ma demonstrates how to use the Sagemaker Python SDK for Object Detection for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` + +.. list-table:: Available Models + :widths: 50 20 20 20 20 + :header-rows: 1 + :class: datatable + + * - Model ID + - Fine Tunable? + - Latest Version + - Min SDK Version + - Source + * - mxnet-od-faster-rcnn-fpn-resnet101-v1d-coco + - False + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-od-faster-rcnn-fpn-resnet50-v1b-coco + - False + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-od-faster-rcnn-resnet101-v1d-coco + - False + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-od-faster-rcnn-resnet50-v1b-coco + - False + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-od-faster-rcnn-resnet50-v1b-voc + - False + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-od-ssd-300-vgg16-atrous-coco + - True + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-od-ssd-300-vgg16-atrous-voc + - True + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-od-ssd-512-mobilenet1-0-coco + - True + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-od-ssd-512-mobilenet1-0-voc + - True + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-od-ssd-512-resnet50-v1-coco + - True + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-od-ssd-512-resnet50-v1-voc + - True + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-od-ssd-512-vgg16-atrous-coco + - True + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-od-ssd-512-vgg16-atrous-voc + - True + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-od-yolo3-darknet53-coco + - False + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-od-yolo3-darknet53-voc + - False + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-od-yolo3-mobilenet1-0-coco + - False + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-od-yolo3-mobilenet1-0-voc + - False + - 2.0.0 + - 2.189.0 + - `GluonCV `__ diff --git a/doc/algorithms/vision/object_detection_pytorch.rst b/doc/algorithms/vision/object_detection_pytorch.rst index aa703e74b5..52791b9d02 100644 --- a/doc/algorithms/vision/object_detection_pytorch.rst +++ b/doc/algorithms/vision/object_detection_pytorch.rst @@ -7,3 +7,19 @@ This is a supervised object detection algorithm which supports fine-tuning of ma demonstrates how to use the Sagemaker Python SDK for Object Detection for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` + +.. list-table:: Available Models + :widths: 50 20 20 20 20 + :header-rows: 1 + :class: datatable + + * - Model ID + - Fine Tunable? + - Latest Version + - Min SDK Version + - Source + * - pytorch-od-nvidia-ssd + - False + - 2.0.0 + - 2.189.0 + - `Pytorch Hub `__ diff --git a/doc/algorithms/vision/object_detection_tensorflow.rst b/doc/algorithms/vision/object_detection_tensorflow.rst index 2536322847..674c8339f1 100644 --- a/doc/algorithms/vision/object_detection_tensorflow.rst +++ b/doc/algorithms/vision/object_detection_tensorflow.rst @@ -7,3 +7,194 @@ This is a supervised object detection algorithm which supports fine-tuning of ma demonstrates how to use the Sagemaker Python SDK for Object Detection for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` + +.. list-table:: Available Models + :widths: 50 20 20 20 20 + :header-rows: 1 + :class: datatable + + * - Model ID + - Fine Tunable? + - Latest Version + - Min SDK Version + - Source + * - tensorflow-od-centernet-hourglass-1024x1024-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-centernet-hourglass-1024x1024-kpts-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-centernet-hourglass-512x512-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-centernet-hourglass-512x512-kpts-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-centernet-resnet101v1-fpn-512x512-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-centernet-resnet50v1-fpn-512x512-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-centernet-resnet50v1-fpn-512x512-kpts-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-centernet-resnet50v2-512x512-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-centernet-resnet50v2-512x512-kpts-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-efficientdet-d0-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-efficientdet-d1-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-efficientdet-d2-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-efficientdet-d3-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-efficientdet-d4-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-efficientdet-d5-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-faster-rcnn-inception-resnet-v2-1024x1024-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-faster-rcnn-inception-resnet-v2-640x640-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-faster-rcnn-resnet101-v1-1024x1024-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-faster-rcnn-resnet101-v1-640x640-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-faster-rcnn-resnet101-v1-800x1333-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-faster-rcnn-resnet152-v1-1024x1024-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-faster-rcnn-resnet152-v1-640x640-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-faster-rcnn-resnet152-v1-800x1333-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-faster-rcnn-resnet50-v1-1024x1024-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-faster-rcnn-resnet50-v1-640x640-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-faster-rcnn-resnet50-v1-800x1333-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-retinanet-resnet101-v1-fpn-1024x1024-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-retinanet-resnet101-v1-fpn-640x640-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-retinanet-resnet152-v1-fpn-1024x1024-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-retinanet-resnet152-v1-fpn-640x640-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-retinanet-resnet50-v1-fpn-1024x1024-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-retinanet-resnet50-v1-fpn-640x640-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-ssd-mobilenet-v1-fpn-640x640-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-ssd-mobilenet-v2-2 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-ssd-mobilenet-v2-fpnlite-320x320-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ + * - tensorflow-od-ssd-mobilenet-v2-fpnlite-640x640-1 + - False + - 3.0.0 + - 2.189.0 + - `Tensorflow Hub `__ diff --git a/doc/algorithms/vision/semantic_segmentation_mxnet.rst b/doc/algorithms/vision/semantic_segmentation_mxnet.rst index b0c60cd560..57917717af 100644 --- a/doc/algorithms/vision/semantic_segmentation_mxnet.rst +++ b/doc/algorithms/vision/semantic_segmentation_mxnet.rst @@ -7,3 +7,34 @@ This is a supervised semantic segmentation algorithm which supports fine-tuning demonstrates how to use the Sagemaker Python SDK for Semantic Segmentation for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` + +.. list-table:: Available Models + :widths: 50 20 20 20 20 + :header-rows: 1 + :class: datatable + + * - Model ID + - Fine Tunable? + - Latest Version + - Min SDK Version + - Source + * - mxnet-semseg-fcn-resnet101-ade + - True + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-semseg-fcn-resnet101-coco + - True + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-semseg-fcn-resnet101-voc + - True + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-semseg-fcn-resnet50-ade + - True + - 2.0.0 + - 2.189.0 + - `GluonCV `__ diff --git a/doc/algorithms/vision/text_embedding_tensorflow_mxnet.rst b/doc/algorithms/vision/text_embedding_tensorflow_mxnet.rst index d015c2ef30..a07a31f008 100644 --- a/doc/algorithms/vision/text_embedding_tensorflow_mxnet.rst +++ b/doc/algorithms/vision/text_embedding_tensorflow_mxnet.rst @@ -7,3 +7,34 @@ This is a supervised text embedding algorithm which supports many pre-trained mo demonstrates how to use the Sagemaker Python SDK for Text Embedding for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` + +.. list-table:: Available Models + :widths: 50 20 20 20 20 + :header-rows: 1 + :class: datatable + + * - Model ID + - Fine Tunable? + - Latest Version + - Min SDK Version + - Source + * - mxnet-tcembedding-robertafin-base-uncased + - False + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-tcembedding-robertafin-base-wiki-uncased + - False + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-tcembedding-robertafin-large-uncased + - False + - 2.0.0 + - 2.189.0 + - `GluonCV `__ + * - mxnet-tcembedding-robertafin-large-wiki-uncased + - False + - 2.0.0 + - 2.189.0 + - `GluonCV `__ diff --git a/tests/unit/sagemaker/jumpstart/model/test_model.py b/tests/unit/sagemaker/jumpstart/model/test_model.py index 7b9d935fb6..da75e5ee28 100644 --- a/tests/unit/sagemaker/jumpstart/model/test_model.py +++ b/tests/unit/sagemaker/jumpstart/model/test_model.py @@ -1933,6 +1933,8 @@ def test_model_display_benchmark_metrics( model.display_benchmark_metrics() + model.display_benchmark_metrics(instance_type="g5.12xlarge") + @mock.patch("sagemaker.jumpstart.model.get_init_kwargs") @mock.patch("sagemaker.jumpstart.utils.verify_model_region_and_return_specs") @mock.patch("sagemaker.jumpstart.model.add_instance_rate_stats_to_benchmark_metrics") From 194de2a5db204ff990d0ec673b7c4d4b33cc906c Mon Sep 17 00:00:00 2001 From: Jonathan Makunga Date: Thu, 30 May 2024 15:22:31 -0700 Subject: [PATCH 09/11] Revert "UT" --- .../text/machine_translation_hugging_face.rst | 41 -- .../text/question_answering_pytorch.rst | 101 --- ...tence_pair_classification_hugging_face.rst | 116 --- ...entence_pair_classification_tensorflow.rst | 61 -- .../text/text_classification_tensorflow.rst | 21 - .../text/text_generation_hugging_face.rst | 96 --- .../text/text_summarization_hugging_face.rst | 51 -- .../vision/image_classification_pytorch.rst | 146 ---- .../image_classification_tensorflow.rst | 686 ------------------ .../vision/image_embedding_tensorflow.rst | 271 ------- .../vision/instance_segmentation_mxnet.rst | 31 - .../vision/object_detection_mxnet.rst | 96 --- .../vision/object_detection_pytorch.rst | 16 - .../vision/object_detection_tensorflow.rst | 191 ----- .../vision/semantic_segmentation_mxnet.rst | 31 - .../text_embedding_tensorflow_mxnet.rst | 31 - .../sagemaker/jumpstart/model/test_model.py | 2 - 17 files changed, 1988 deletions(-) diff --git a/doc/algorithms/text/machine_translation_hugging_face.rst b/doc/algorithms/text/machine_translation_hugging_face.rst index e6a8ab6893..d533d0e64d 100644 --- a/doc/algorithms/text/machine_translation_hugging_face.rst +++ b/doc/algorithms/text/machine_translation_hugging_face.rst @@ -8,44 +8,3 @@ This is a supervised machine translation algorithm which supports many pre-train demonstrates how to use the Sagemaker Python SDK for Machine Translation for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK `. - -.. list-table:: Available Models - :widths: 50 20 20 20 20 - :header-rows: 1 - :class: datatable - - * - Model ID - - Fine Tunable? - - Latest Version - - Min SDK Version - - Source - * - huggingface-translation-opus-mt-en-es - - False - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-translation-opus-mt-en-vi - - False - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-translation-opus-mt-mul-en - - False - - 1.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-translation-t5-base - - False - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-translation-t5-large - - False - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-translation-t5-small - - False - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ diff --git a/doc/algorithms/text/question_answering_pytorch.rst b/doc/algorithms/text/question_answering_pytorch.rst index 4ad2205f40..9d9d74ccb1 100644 --- a/doc/algorithms/text/question_answering_pytorch.rst +++ b/doc/algorithms/text/question_answering_pytorch.rst @@ -7,104 +7,3 @@ This is a supervised question answering algorithm which supports fine-tuning of demonstrates how to use the Sagemaker Python SDK for Question Answering for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` - -.. list-table:: Available Models - :widths: 50 20 20 20 20 - :header-rows: 1 - :class: datatable - - * - Model ID - - Fine Tunable? - - Latest Version - - Min SDK Version - - Source - * - pytorch-eqa-bert-base-cased - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ - * - pytorch-eqa-bert-base-multilingual-cased - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ - * - pytorch-eqa-bert-base-multilingual-uncased - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ - * - pytorch-eqa-bert-base-uncased - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ - * - pytorch-eqa-bert-large-cased - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ - * - pytorch-eqa-bert-large-cased-whole-word-masking - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ - * - pytorch-eqa-bert-large-cased-whole-word-masking-finetuned-squad - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ - * - pytorch-eqa-bert-large-uncased - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ - * - pytorch-eqa-bert-large-uncased-whole-word-masking - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ - * - pytorch-eqa-bert-large-uncased-whole-word-masking-finetuned-squad - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ - * - pytorch-eqa-distilbert-base-cased - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ - * - pytorch-eqa-distilbert-base-multilingual-cased - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ - * - pytorch-eqa-distilbert-base-uncased - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ - * - pytorch-eqa-distilroberta-base - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ - * - pytorch-eqa-roberta-base - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ - * - pytorch-eqa-roberta-base-openai-detector - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ - * - pytorch-eqa-roberta-large - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ - * - pytorch-eqa-roberta-large-openai-detector - - True - - 1.2.1 - - 2.75.0 - - `Pytorch Hub `__ diff --git a/doc/algorithms/text/sentence_pair_classification_hugging_face.rst b/doc/algorithms/text/sentence_pair_classification_hugging_face.rst index a96bb9fac0..2892b9d516 100644 --- a/doc/algorithms/text/sentence_pair_classification_hugging_face.rst +++ b/doc/algorithms/text/sentence_pair_classification_hugging_face.rst @@ -7,119 +7,3 @@ This is a supervised sentence pair classification algorithm which supports fine- demonstrates how to use the Sagemaker Python SDK for Sentence Pair Classification for using these algorithms. For detailed documentation please refer `Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK `__ - -.. list-table:: Available Models - :widths: 50 20 20 20 20 - :header-rows: 1 - :class: datatable - - * - Model ID - - Fine Tunable? - - Latest Version - - Min SDK Version - - Source - * - huggingface-spc-bert-base-cased - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-bert-base-multilingual-cased - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-bert-base-multilingual-uncased - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-bert-base-uncased - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-bert-large-cased - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-bert-large-cased-whole-word-masking - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-bert-large-uncased - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-bert-large-uncased-whole-word-masking - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-distilbert-base-cased - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-distilbert-base-multilingual-cased - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-distilbert-base-uncased - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-distilroberta-base - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-roberta-base - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-roberta-base-openai-detector - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-roberta-large - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-roberta-large-openai-detector - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-xlm-clm-ende-1024 - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-xlm-mlm-ende-1024 - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-xlm-mlm-enro-1024 - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-xlm-mlm-tlm-xnli15-1024 - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-spc-xlm-mlm-xnli15-1024 - - True - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ diff --git a/doc/algorithms/text/sentence_pair_classification_tensorflow.rst b/doc/algorithms/text/sentence_pair_classification_tensorflow.rst index 7b4a079dd5..80264e84f3 100644 --- a/doc/algorithms/text/sentence_pair_classification_tensorflow.rst +++ b/doc/algorithms/text/sentence_pair_classification_tensorflow.rst @@ -7,64 +7,3 @@ This is a supervised sentence pair classification algorithm which supports fine- demonstrates how to use the Sagemaker Python SDK for Sentence Pair Classification for using these algorithms. For detailed documentation please refer `Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK `__ - -.. list-table:: Available Models - :widths: 50 20 20 20 20 - :header-rows: 1 - :class: datatable - - * - Model ID - - Fine Tunable? - - Latest Version - - Min SDK Version - - Source - * - tensorflow-spc-bert-en-cased-L-12-H-768-A-12-2 - - True - - 2.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-spc-bert-en-uncased-L-12-H-768-A-12-2 - - True - - 2.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-spc-bert-en-uncased-L-24-H-1024-A-16-2 - - True - - 2.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-spc-bert-en-wwm-cased-L-24-H-1024-A-16-2 - - True - - 2.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-spc-bert-en-wwm-uncased-L-24-H-1024-A-16-2 - - True - - 2.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-spc-bert-multi-cased-L-12-H-768-A-12-2 - - True - - 2.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-spc-electra-base-1 - - True - - 2.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-spc-electra-small-1 - - True - - 2.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-spc-experts-bert-pubmed-1 - - True - - 2.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-spc-experts-bert-wiki-books-1 - - True - - 2.0.0 - - 2.189.0 - - `Tensorflow Hub `__ diff --git a/doc/algorithms/text/text_classification_tensorflow.rst b/doc/algorithms/text/text_classification_tensorflow.rst index 3536556bb8..c60a5b3e1c 100644 --- a/doc/algorithms/text/text_classification_tensorflow.rst +++ b/doc/algorithms/text/text_classification_tensorflow.rst @@ -7,24 +7,3 @@ This is a supervised text classification algorithm which supports fine-tuning of demonstrates how to use the Sagemaker Python SDK for Text Classification for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` - -.. list-table:: Available Models - :widths: 50 20 20 20 20 - :header-rows: 1 - :class: datatable - - * - Model ID - - Fine Tunable? - - Latest Version - - Min SDK Version - - Source - * - tensorflow-tc-albert-en-base - - True - - 2.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-tc-bert-en-cased-L-12-H-768-A-12-2 - - True - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ diff --git a/doc/algorithms/text/text_generation_hugging_face.rst b/doc/algorithms/text/text_generation_hugging_face.rst index a865f75c65..30fae26196 100644 --- a/doc/algorithms/text/text_generation_hugging_face.rst +++ b/doc/algorithms/text/text_generation_hugging_face.rst @@ -7,99 +7,3 @@ This is a supervised text generation algorithm which supports many pre-trained m demonstrates how to use the Sagemaker Python SDK for Text Generation for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` - -.. list-table:: Available Models - :widths: 50 20 20 20 20 - :header-rows: 1 - :class: datatable - - * - Model ID - - Fine Tunable? - - Latest Version - - Min SDK Version - - Source - * - huggingface-textgeneration-bloom-1b1 - - False - - 2.1.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-textgeneration-bloom-1b7 - - False - - 2.1.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-textgeneration-bloom-560m - - False - - 2.1.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-textgeneration-bloomz-1b1 - - False - - 2.1.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-textgeneration-bloomz-1b7 - - False - - 2.1.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-textgeneration-bloomz-560m - - False - - 2.1.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-textgeneration-distilgpt2 - - False - - 2.1.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-textgeneration-dolly-v2-12b-bf16 - - False - - 2.1.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-textgeneration-dolly-v2-3b-bf16 - - False - - 2.1.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-textgeneration-dolly-v2-7b-bf16 - - False - - 2.1.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-textgeneration-falcon-40b-bf16 - - False - - 1.0.0 - - 2.144.0 - - `HuggingFace `__ - * - huggingface-textgeneration-falcon-40b-instruct-bf16 - - False - - 1.0.0 - - 2.144.0 - - `HuggingFace `__ - * - huggingface-textgeneration-falcon-7b-bf16 - - False - - 1.0.0 - - 2.144.0 - - `HuggingFace `__ - * - huggingface-textgeneration-falcon-7b-instruct-bf16 - - False - - 1.0.0 - - 2.144.0 - - `HuggingFace `__ - * - huggingface-textgeneration-gpt2 - - False - - 2.1.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-textgeneration-models - - False - - 1.3.0 - - 2.144.0 - - `HuggingFace `__ - * - huggingface-textgeneration-open-llama - - False - - 3.1.0 - - 2.189.0 - - `HuggingFace `__ diff --git a/doc/algorithms/text/text_summarization_hugging_face.rst b/doc/algorithms/text/text_summarization_hugging_face.rst index af0346b146..206c880ba3 100644 --- a/doc/algorithms/text/text_summarization_hugging_face.rst +++ b/doc/algorithms/text/text_summarization_hugging_face.rst @@ -7,54 +7,3 @@ This is a supervised text summarization algorithm which supports many pre-traine demonstrates how to use the Sagemaker Python SDK for Text Summarization for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` - -.. list-table:: Available Models - :widths: 50 20 20 20 20 - :header-rows: 1 - :class: datatable - - * - Model ID - - Fine Tunable? - - Latest Version - - Min SDK Version - - Source - * - huggingface-summarization-bart-large-cnn-samsum - - False - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-summarization-bert-small2bert-small-finetuned-cnn-daily-mail-summarization - - False - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-summarization-bigbird-pegasus-large-arxiv - - False - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-summarization-bigbird-pegasus-large-pubmed - - False - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-summarization-distilbart-cnn-12-6 - - False - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-summarization-distilbart-cnn-6-6 - - False - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-summarization-distilbart-xsum-1-1 - - False - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ - * - huggingface-summarization-distilbart-xsum-12-3 - - False - - 2.0.0 - - 2.189.0 - - `HuggingFace `__ diff --git a/doc/algorithms/vision/image_classification_pytorch.rst b/doc/algorithms/vision/image_classification_pytorch.rst index 88a8e67571..3c154c6cfe 100644 --- a/doc/algorithms/vision/image_classification_pytorch.rst +++ b/doc/algorithms/vision/image_classification_pytorch.rst @@ -7,149 +7,3 @@ This is a supervised image clasification algorithm which supports fine-tuning of demonstrates how to use the Sagemaker Python SDK for Image Classification for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` - -.. list-table:: Available Models - :widths: 50 20 20 20 20 - :header-rows: 1 - :class: datatable - - * - Model ID - - Fine Tunable? - - Latest Version - - Min SDK Version - - Source - * - pytorch-ic-alexnet - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-densenet121 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-densenet161 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-densenet169 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-densenet201 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-googlenet - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-mobilenet-v2 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-resnet101 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-resnet152 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-resnet18 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-resnet34 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-resnet50 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-resnext101-32x8d - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-resnext50-32x4d - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-shufflenet-v2-x1-0 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-squeezenet1-0 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-squeezenet1-1 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-vgg11 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-vgg11-bn - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-vgg13 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-vgg13-bn - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-vgg16 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-vgg16-bn - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-vgg19 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-vgg19-bn - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-wide-resnet101-2 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ - * - pytorch-ic-wide-resnet50-2 - - True - - 3.0.0 - - 2.189.0 - - `Pytorch Hub `__ diff --git a/doc/algorithms/vision/image_classification_tensorflow.rst b/doc/algorithms/vision/image_classification_tensorflow.rst index 5f6cd309be..e49820ee50 100644 --- a/doc/algorithms/vision/image_classification_tensorflow.rst +++ b/doc/algorithms/vision/image_classification_tensorflow.rst @@ -7,689 +7,3 @@ This is a supervised image clasification algorithm which supports fine-tuning of demonstrates how to use the Sagemaker Python SDK for Image Classification for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` - -.. list-table:: Available Models - :widths: 50 20 20 20 20 - :header-rows: 1 - :class: datatable - - * - Model ID - - Fine Tunable? - - Latest Version - - Min SDK Version - - Source - * - tensorflow-ic-bit-m-r101x1-ilsvrc2012-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-bit-m-r101x1-imagenet21k-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-bit-m-r101x3-ilsvrc2012-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-bit-m-r101x3-imagenet21k-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-bit-m-r152x4-ilsvrc2012 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-bit-m-r152x4-imagenet21k - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-bit-m-r50x1-ilsvrc2012-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-bit-m-r50x1-imagenet21k-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-bit-m-r50x3-ilsvrc2012-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-bit-m-r50x3-imagenet21k-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-bit-s-r101x1-ilsvrc2012-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-bit-s-r101x3-ilsvrc2012-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-bit-s-r152x4-ilsvrc2012 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-bit-s-r50x1-ilsvrc2012-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-bit-s-r50x3-ilsvrc2012-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-cait-m36-384 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-cait-m48-448 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-cait-s24-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-cait-s24-384 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-cait-s36-384 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-cait-xs24-384 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-cait-xxs24-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-cait-xxs24-384 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-cait-xxs36-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-cait-xxs36-384 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-deit-base-distilled-patch16-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-deit-base-distilled-patch16-384 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-deit-base-patch16-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-deit-base-patch16-384 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-deit-small-distilled-patch16-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-deit-small-patch16-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-deit-tiny-distilled-patch16-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-deit-tiny-patch16-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-b0-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-b1-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-b2-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-b3-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-b4-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-b5-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-b6-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-b7-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-lite0-classification-2 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-lite1-classification-2 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-lite2-classification-2 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-lite3-classification-2 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-lite4-classification-2 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet1k-b0 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet1k-b1 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet1k-b2 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet1k-b3 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet1k-l - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet1k-m - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet1k-s - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet21k-b0 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet21k-b1 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet21k-b2 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet21k-b3 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-b0 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-b1 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-b2 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-b3 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-l - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-m - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-s - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-xl - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet21k-l - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet21k-m - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet21k-s - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-efficientnet-v2-imagenet21k-xl - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-inception-resnet-v2-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-inception-v1-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-inception-v2-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-inception-v3-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v1-025-128-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v1-025-160-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v1-025-192-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v1-025-224-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v1-050-128-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v1-050-160-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v1-050-192-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v1-050-224-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v1-075-128-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v1-075-160-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v1-075-192-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v1-075-224-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v1-100-128-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v1-100-160-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v1-100-192-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v1-100-224-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-035-128 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-035-160 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-035-192 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-035-224-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-035-96 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-050-128 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-050-160 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-050-192 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-050-224-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-050-96 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-075-128 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-075-160 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-075-192 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-075-224-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-075-96 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-100-160 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-100-192 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-100-224-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-100-96 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-130-224-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v2-140-224-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v3-large-075-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v3-large-100-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v3-small-075-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-mobilenet-v3-small-100-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-nasnet-large - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-nasnet-mobile - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-pnasnet-large - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-resnet-v1-101-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-resnet-v1-152-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-resnet-v1-50-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-resnet-v2-101-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-resnet-v2-152-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-imagenet-resnet-v2-50-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-resnet-50-classification-1 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-swin-base-patch4-window12-384 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-swin-base-patch4-window7-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-swin-large-patch4-window12-384 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-swin-large-patch4-window7-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-swin-s3-base-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-swin-s3-small-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-swin-s3-tiny-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-swin-small-patch4-window7-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-swin-tiny-patch4-window7-224 - - True - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-tf2-preview-inception-v3-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-ic-tf2-preview-mobilenet-v2-classification-4 - - True - - 4.0.2 - - 2.189.0 - - `Tensorflow Hub `__ diff --git a/doc/algorithms/vision/image_embedding_tensorflow.rst b/doc/algorithms/vision/image_embedding_tensorflow.rst index bfba13686c..0938377354 100644 --- a/doc/algorithms/vision/image_embedding_tensorflow.rst +++ b/doc/algorithms/vision/image_embedding_tensorflow.rst @@ -7,274 +7,3 @@ This is a supervised image embedding algorithm which supports many pre-trained m demonstrates how to use the Sagemaker Python SDK for Image Embedding for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` - -.. list-table:: Available Models - :widths: 50 20 20 20 20 - :header-rows: 1 - :class: datatable - - * - Model ID - - Fine Tunable? - - Latest Version - - Min SDK Version - - Source - * - tensorflow-icembedding-bit-m-r101x1-ilsvrc2012-featurevector-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-bit-m-r101x3-imagenet21k-featurevector-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-bit-m-r50x1-ilsvrc2012-featurevector-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-bit-m-r50x3-imagenet21k-featurevector-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-bit-s-r101x1-ilsvrc2012-featurevector-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-bit-s-r101x3-ilsvrc2012-featurevector-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-bit-s-r50x1-ilsvrc2012-featurevector-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-bit-s-r50x3-ilsvrc2012-featurevector-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-efficientnet-b0-featurevector-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-efficientnet-b1-featurevector-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-efficientnet-b2-featurevector-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-efficientnet-b3-featurevector-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-efficientnet-b6-featurevector-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-efficientnet-lite0-featurevector-2 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-efficientnet-lite1-featurevector-2 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-efficientnet-lite2-featurevector-2 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-efficientnet-lite3-featurevector-2 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-efficientnet-lite4-featurevector-2 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-inception-v1-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-inception-v2-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-inception-v3-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v1-025-128-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v1-025-160-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v1-025-192-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v1-025-224-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v1-050-128-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v1-050-160-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v1-050-192-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v1-050-224-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v1-075-128-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v1-075-160-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v1-075-192-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v1-075-224-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v1-100-128-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v1-100-160-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v1-100-192-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v1-100-224-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v2-035-224-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v2-050-224-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v2-075-224-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v2-100-224-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v2-130-224-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-mobilenet-v2-140-224-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-resnet-v1-101-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-resnet-v1-152-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-resnet-v1-50-featurevector-4 - - False - - 2.0.2 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-resnet-v2-101-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-resnet-v2-152-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-imagenet-resnet-v2-50-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-resnet-50-featurevector-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-tf2-preview-inception-v3-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-icembedding-tf2-preview-mobilenet-v2-featurevector-4 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ diff --git a/doc/algorithms/vision/instance_segmentation_mxnet.rst b/doc/algorithms/vision/instance_segmentation_mxnet.rst index 33aa87fe28..a38611bc9a 100644 --- a/doc/algorithms/vision/instance_segmentation_mxnet.rst +++ b/doc/algorithms/vision/instance_segmentation_mxnet.rst @@ -7,34 +7,3 @@ This is a supervised image segmentation algorithm which supports many pre-train demonstrates how to use the Sagemaker Python SDK for Image Segmentation for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` - -.. list-table:: Available Models - :widths: 50 20 20 20 20 - :header-rows: 1 - :class: datatable - - * - Model ID - - Fine Tunable? - - Latest Version - - Min SDK Version - - Source - * - mxnet-is-mask-rcnn-fpn-resnet101-v1d-coco - - False - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-is-mask-rcnn-fpn-resnet18-v1b-coco - - False - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-is-mask-rcnn-fpn-resnet50-v1b-coco - - False - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-is-mask-rcnn-resnet18-v1b-coco - - False - - 2.0.0 - - 2.189.0 - - `GluonCV `__ diff --git a/doc/algorithms/vision/object_detection_mxnet.rst b/doc/algorithms/vision/object_detection_mxnet.rst index 70d6b55178..9ce52f992b 100644 --- a/doc/algorithms/vision/object_detection_mxnet.rst +++ b/doc/algorithms/vision/object_detection_mxnet.rst @@ -7,99 +7,3 @@ This is a supervised object detection algorithm which supports fine-tuning of ma demonstrates how to use the Sagemaker Python SDK for Object Detection for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` - -.. list-table:: Available Models - :widths: 50 20 20 20 20 - :header-rows: 1 - :class: datatable - - * - Model ID - - Fine Tunable? - - Latest Version - - Min SDK Version - - Source - * - mxnet-od-faster-rcnn-fpn-resnet101-v1d-coco - - False - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-od-faster-rcnn-fpn-resnet50-v1b-coco - - False - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-od-faster-rcnn-resnet101-v1d-coco - - False - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-od-faster-rcnn-resnet50-v1b-coco - - False - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-od-faster-rcnn-resnet50-v1b-voc - - False - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-od-ssd-300-vgg16-atrous-coco - - True - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-od-ssd-300-vgg16-atrous-voc - - True - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-od-ssd-512-mobilenet1-0-coco - - True - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-od-ssd-512-mobilenet1-0-voc - - True - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-od-ssd-512-resnet50-v1-coco - - True - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-od-ssd-512-resnet50-v1-voc - - True - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-od-ssd-512-vgg16-atrous-coco - - True - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-od-ssd-512-vgg16-atrous-voc - - True - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-od-yolo3-darknet53-coco - - False - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-od-yolo3-darknet53-voc - - False - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-od-yolo3-mobilenet1-0-coco - - False - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-od-yolo3-mobilenet1-0-voc - - False - - 2.0.0 - - 2.189.0 - - `GluonCV `__ diff --git a/doc/algorithms/vision/object_detection_pytorch.rst b/doc/algorithms/vision/object_detection_pytorch.rst index 52791b9d02..aa703e74b5 100644 --- a/doc/algorithms/vision/object_detection_pytorch.rst +++ b/doc/algorithms/vision/object_detection_pytorch.rst @@ -7,19 +7,3 @@ This is a supervised object detection algorithm which supports fine-tuning of ma demonstrates how to use the Sagemaker Python SDK for Object Detection for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` - -.. list-table:: Available Models - :widths: 50 20 20 20 20 - :header-rows: 1 - :class: datatable - - * - Model ID - - Fine Tunable? - - Latest Version - - Min SDK Version - - Source - * - pytorch-od-nvidia-ssd - - False - - 2.0.0 - - 2.189.0 - - `Pytorch Hub `__ diff --git a/doc/algorithms/vision/object_detection_tensorflow.rst b/doc/algorithms/vision/object_detection_tensorflow.rst index 674c8339f1..2536322847 100644 --- a/doc/algorithms/vision/object_detection_tensorflow.rst +++ b/doc/algorithms/vision/object_detection_tensorflow.rst @@ -7,194 +7,3 @@ This is a supervised object detection algorithm which supports fine-tuning of ma demonstrates how to use the Sagemaker Python SDK for Object Detection for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` - -.. list-table:: Available Models - :widths: 50 20 20 20 20 - :header-rows: 1 - :class: datatable - - * - Model ID - - Fine Tunable? - - Latest Version - - Min SDK Version - - Source - * - tensorflow-od-centernet-hourglass-1024x1024-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-centernet-hourglass-1024x1024-kpts-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-centernet-hourglass-512x512-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-centernet-hourglass-512x512-kpts-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-centernet-resnet101v1-fpn-512x512-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-centernet-resnet50v1-fpn-512x512-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-centernet-resnet50v1-fpn-512x512-kpts-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-centernet-resnet50v2-512x512-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-centernet-resnet50v2-512x512-kpts-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-efficientdet-d0-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-efficientdet-d1-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-efficientdet-d2-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-efficientdet-d3-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-efficientdet-d4-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-efficientdet-d5-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-faster-rcnn-inception-resnet-v2-1024x1024-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-faster-rcnn-inception-resnet-v2-640x640-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-faster-rcnn-resnet101-v1-1024x1024-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-faster-rcnn-resnet101-v1-640x640-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-faster-rcnn-resnet101-v1-800x1333-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-faster-rcnn-resnet152-v1-1024x1024-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-faster-rcnn-resnet152-v1-640x640-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-faster-rcnn-resnet152-v1-800x1333-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-faster-rcnn-resnet50-v1-1024x1024-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-faster-rcnn-resnet50-v1-640x640-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-faster-rcnn-resnet50-v1-800x1333-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-retinanet-resnet101-v1-fpn-1024x1024-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-retinanet-resnet101-v1-fpn-640x640-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-retinanet-resnet152-v1-fpn-1024x1024-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-retinanet-resnet152-v1-fpn-640x640-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-retinanet-resnet50-v1-fpn-1024x1024-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-retinanet-resnet50-v1-fpn-640x640-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-ssd-mobilenet-v1-fpn-640x640-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-ssd-mobilenet-v2-2 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-ssd-mobilenet-v2-fpnlite-320x320-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ - * - tensorflow-od-ssd-mobilenet-v2-fpnlite-640x640-1 - - False - - 3.0.0 - - 2.189.0 - - `Tensorflow Hub `__ diff --git a/doc/algorithms/vision/semantic_segmentation_mxnet.rst b/doc/algorithms/vision/semantic_segmentation_mxnet.rst index 57917717af..b0c60cd560 100644 --- a/doc/algorithms/vision/semantic_segmentation_mxnet.rst +++ b/doc/algorithms/vision/semantic_segmentation_mxnet.rst @@ -7,34 +7,3 @@ This is a supervised semantic segmentation algorithm which supports fine-tuning demonstrates how to use the Sagemaker Python SDK for Semantic Segmentation for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` - -.. list-table:: Available Models - :widths: 50 20 20 20 20 - :header-rows: 1 - :class: datatable - - * - Model ID - - Fine Tunable? - - Latest Version - - Min SDK Version - - Source - * - mxnet-semseg-fcn-resnet101-ade - - True - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-semseg-fcn-resnet101-coco - - True - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-semseg-fcn-resnet101-voc - - True - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-semseg-fcn-resnet50-ade - - True - - 2.0.0 - - 2.189.0 - - `GluonCV `__ diff --git a/doc/algorithms/vision/text_embedding_tensorflow_mxnet.rst b/doc/algorithms/vision/text_embedding_tensorflow_mxnet.rst index a07a31f008..d015c2ef30 100644 --- a/doc/algorithms/vision/text_embedding_tensorflow_mxnet.rst +++ b/doc/algorithms/vision/text_embedding_tensorflow_mxnet.rst @@ -7,34 +7,3 @@ This is a supervised text embedding algorithm which supports many pre-trained mo demonstrates how to use the Sagemaker Python SDK for Text Embedding for using these algorithms. For detailed documentation please refer :ref:`Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK ` - -.. list-table:: Available Models - :widths: 50 20 20 20 20 - :header-rows: 1 - :class: datatable - - * - Model ID - - Fine Tunable? - - Latest Version - - Min SDK Version - - Source - * - mxnet-tcembedding-robertafin-base-uncased - - False - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-tcembedding-robertafin-base-wiki-uncased - - False - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-tcembedding-robertafin-large-uncased - - False - - 2.0.0 - - 2.189.0 - - `GluonCV `__ - * - mxnet-tcembedding-robertafin-large-wiki-uncased - - False - - 2.0.0 - - 2.189.0 - - `GluonCV `__ diff --git a/tests/unit/sagemaker/jumpstart/model/test_model.py b/tests/unit/sagemaker/jumpstart/model/test_model.py index da75e5ee28..7b9d935fb6 100644 --- a/tests/unit/sagemaker/jumpstart/model/test_model.py +++ b/tests/unit/sagemaker/jumpstart/model/test_model.py @@ -1933,8 +1933,6 @@ def test_model_display_benchmark_metrics( model.display_benchmark_metrics() - model.display_benchmark_metrics(instance_type="g5.12xlarge") - @mock.patch("sagemaker.jumpstart.model.get_init_kwargs") @mock.patch("sagemaker.jumpstart.utils.verify_model_region_and_return_specs") @mock.patch("sagemaker.jumpstart.model.add_instance_rate_stats_to_benchmark_metrics") From 5957f17c1eabcad5ab2c3c11af53de300099a32a Mon Sep 17 00:00:00 2001 From: Jonathan Makunga Date: Thu, 30 May 2024 15:23:42 -0700 Subject: [PATCH 10/11] UT --- tests/unit/sagemaker/jumpstart/model/test_model.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/unit/sagemaker/jumpstart/model/test_model.py b/tests/unit/sagemaker/jumpstart/model/test_model.py index 7b9d935fb6..1e25aff20b 100644 --- a/tests/unit/sagemaker/jumpstart/model/test_model.py +++ b/tests/unit/sagemaker/jumpstart/model/test_model.py @@ -1932,6 +1932,7 @@ def test_model_display_benchmark_metrics( model = JumpStartModel(model_id=model_id) model.display_benchmark_metrics() + model.display_benchmark_metrics("g5.12xlarge") @mock.patch("sagemaker.jumpstart.model.get_init_kwargs") @mock.patch("sagemaker.jumpstart.utils.verify_model_region_and_return_specs") From 1f86e81d35ff7026ff93b8caa40f3bf9f5d3e407 Mon Sep 17 00:00:00 2001 From: Jonathan Makunga Date: Thu, 30 May 2024 15:55:26 -0700 Subject: [PATCH 11/11] UT --- tests/unit/sagemaker/jumpstart/model/test_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/sagemaker/jumpstart/model/test_model.py b/tests/unit/sagemaker/jumpstart/model/test_model.py index 1e25aff20b..e80ce020f7 100644 --- a/tests/unit/sagemaker/jumpstart/model/test_model.py +++ b/tests/unit/sagemaker/jumpstart/model/test_model.py @@ -1932,7 +1932,7 @@ def test_model_display_benchmark_metrics( model = JumpStartModel(model_id=model_id) model.display_benchmark_metrics() - model.display_benchmark_metrics("g5.12xlarge") + model.display_benchmark_metrics(instance_type="g5.12xlarge") @mock.patch("sagemaker.jumpstart.model.get_init_kwargs") @mock.patch("sagemaker.jumpstart.utils.verify_model_region_and_return_specs")