是否可以在没有SageMaker training toolkit的情况下进行SageMaker超参数调优任务？

Question

【以下的问题经过翻译处理】 我已经创建了自定义的训练镜像，可以通过CreateTrainingJob进行操作，没有使用sagemaker training toolkit（需要“ContainerEntrypoint”选项）。

但是当我尝试运行HyperParameter Tuning Job时，在“AlgorithmSpecification”字段中不允许添加“ContainerEntrypoint”选项。

不使用 sagemaker training toolkit的训练镜像是否能运行超参调优任务？

谢谢！

Answer

【以下的回答经过翻译处理】 你好！

要使用自定义训练镜像，可以在Dockerfile中指定Entrypoint。

以下是一些代码片段和链接，可供参考：

```
ENTRYPOINT ["./entrypoint.sh"]
```

```
ENTRYPOINT ["python", "-m", "trainer.start"]
```

此外，SageMaker Training Toolkit是一个精心包装的Python包，可帮助您创建自定义训练图像，这与自己实现逻辑没有任何区别。

因此，使用自定义容器运行超参数调整作业而不使用SageMaker Training Toolkit绝对是可能的。

相关内容