sagemaker-training-toolkit

把训练脚本塞进容器，SageMaker 自动跑训练并上传模型

开源 free 模型训练

访问官网 ↗ GitHub ↗ 文档 ↗

GitHub 星标 ★ 530

维护状态活跃

是否开源是

定价模式 free

项目预览

项目数据

分类模型训练

开发团队aws

所属国家

官网地址

GitHubhttps://github.com/aws/sagemaker-training-toolkit

定价模式free

价格说明开源免费，Apache 2.0 许可证

访问状态

是否开源是

开源协议Apache-2.0

主要语言Python

技术栈/模型aws,deep-learning,docker,machine-learning,python,sagemaker,training

GitHub 星标★ 530

30天Star增速

HF 下载量

上线时间2019-10-02 00:00:00

代码更新—

仓库体积—

本站数据更新—

维护状态活跃

中文支持

访问方式

移动端支持

综合评分

收录时间2026-10-12

浏览次数0

项目介绍

SageMaker Training Toolkit 是 AWS 官方开源的 Python 库，用于在 Docker 容器内训练机器学习模型并接入 Amazon SageMaker。它解决的核心问题是：用户无需自己编写容器与 SageMaker 服务之间的通信代码，只需把训练脚本按约定放入容器，工具包会自动读取 SageMaker 注入的环境变量（如超参数、数据通道、模型输出路径），完成数据下载、训练启动、指标上报和模型打包。核心能力包括：解析 SM_* 环境变量、支持多数据通道与分布式训练、自动上传模型产物到 S3、兼容 TensorFlow/PyTorch/MXNet 等框架的官方容器。它让本地开发的训练脚本几乎零改动迁移到 SageMaker 托管训练，是构建自定义训练容器的标准底座。