特定于任务的模型 - Amazon SageMaker
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅 中国的 Amazon Web Services 服务入门 (PDF)

本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。

特定于任务的模型

JumpStart 支持十五种最流行的问题类型的任务特定模型。在支持的问题类型中,共有十三种类型与视觉和 NLP 相关。有八种问题类型支持增量训练和微调。有关增量训练和超参数调整的更多信息,请参阅SageMaker 自动模型调整。 JumpStart 还支持四种常用的表格数据建模算法。

您可以从 Studio 或 Studio Classic 的 JumpStart 登录页面搜索和浏览模特。当您选择模型时,模型详细信息页面会提供有关该模型的信息,您可以通过几个步骤来训练和部署模型。描述部分介绍了您可以通过模型完成的任务、预期的输入和输出类型以及微调模型所需的数据类型。

您还可以通过 SageMaker Python 软件开发工具包以编程方式使用模型。有关所有可用型号的列表,请参阅JumpStart可用型号表

下表汇总了问题类型列表及其示例 Jupyter 笔记本的链接。

问题类型 支持使用预训练模型进行推理 可在自定义数据集上训练 支持的框架 示例笔记本
图像分类

PyTorch, TensorFlow

简介 JumpStart -图像分类

对象检测 PyTorch, TensorFlow,mxNet

简介 JumpStart -物体检测

语义分割 MXNet

简介 JumpStart -语义分割

实例分段 MXNet

简介 JumpStart -实例分割

图像嵌入 不支持 TensorFlow,mxNet

简介 JumpStart -图像嵌入

文本分类 TensorFlow

简介 JumpStart -文本分类

句子对分类 TensorFlow,Hugging Face

简介 JumpStart -句子对分类

问题回答 PyTorch,Hugging Face

简介 JumpStart — 问题解答

指定实体识别 不支持 Hugging Face

简介 JumpStart -命名实体识别

文本摘要 不支持 Hugging Face

简介 JumpStart -文本摘要

文本生成 不支持 Hugging Face

简介 JumpStart -文本生成

机器翻译 不支持 Hugging Face

简介 JumpStart -机器翻译

文本嵌入 不支持 TensorFlow,mxNet

简介 JumpStart -文本嵌入

表格分类 LightGBM、、xgBoost CatBoost、 AutoGluon-Tabular、、Linear Learner TabTransformer

简介 JumpStart ——表格分类——LightGBM, CatBoost

简介 JumpStart ——表格分类——xgBoost,Linear Learner

简介 JumpStart -表格分类-学员 AutoGluon

简介 JumpStart -表格分类-学员 TabTransformer

表格回归 LightGBM、、xgBoost CatBoost、 AutoGluon-Tabular、、Linear Learner TabTransformer

简介 JumpStart ——表格回归——lightGBM, CatBoost

简介 JumpStart — 表格回归——xgBoost,Linear Learner

简介 JumpStart — 表格回归-学员 AutoGluon

简介 JumpStart — 表格回归-学员 TabTransformer