Amazon SageMaker 集成

我们提供了一个简单的管道，使您能够以最小的努力在Amazon SageMaker上运行Fortuna。

  1. 创建一个AWS账户 - 这是免费的！保存账户ID和你想要启动训练任务的区域。

  2. 首先，更新您的本地AWS凭证。 然后您需要构建并将Docker镜像推送到Amazon ECR仓库。 这个脚本将帮助您完成这一过程 - 它将需要您的AWS账户ID和区域。如果您需要在Docker镜像中包含其他包， 您应该考虑自定义Dockerfile。 注意：该脚本已在M1 MacOS上测试。 不同的操作系统可能需要进行小的修改。

  3. 创建一个S3 bucket。 你需要这个来在Amazon Sagemaker上转储训练任务的结果。

  1. 编写一个配置yaml文件。这将包括您的AWS详细信息、您想在Amazon SageMaker上运行的入口点脚本的路径、传递给脚本的参数、您想要转储结果的S3存储桶的路径、要监控的指标等。请参阅这里查看示例。

  2. 最后，给定 config_dir，即主配置目录的绝对路径， 和 config_filename，即主配置文件的名称（不带 .yaml 扩展名）， 进入 Python 并运行以下内容：

from fortuna.sagemaker import run_training_job
run_training_job(config_dir=config_dir, config_filename=config_filename)