运行SageMaker训练作业时能否在自定义容器内运行自定义脚本

自定义容器不存在该使用限制，完全支持复用同一镜像运行指定训练脚本，实现和内置算法一致的使用体验。

• 你提到的「创建训练作业默认执行镜像内预置train文件」只是SageMaker的兜底默认规则，不是强制约束。这个规则的触发前提是你没有给容器配置自定义入口逻辑，也没有适配SageMaker的启动参数解析规则。
• 要实现动态指定运行脚本，有两种成熟的落地方式：
  1. 基于SageMaker官方训练工具包实现（最省心，和内置算法体验完全对齐）
     构建自定义镜像时预装sagemaker-training工具包，将镜像的ENTRYPOINT设置为工具包提供的启动命令即可。后续创建训练作业时，你可以像使用内置算法一样，直接在Estimator配置里传入entry_point（要运行的自定义脚本路径）、source_dir（脚本依赖的代码目录）参数，工具包会自动完成代码下载、依赖安装、脚本执行的全流程，不需要在镜像内提前预置任何训练代码，同一个镜像可以支撑任意符合依赖要求的训练任务。
  2. 自定义入口脚本实现
     如果你不想引入官方工具包，可以自行编写容器的入口启动脚本：SageMaker启动训练容器时，会把你通过API/SDK传入的代码包自动解压挂载到容器内/opt/ml/input/data/code/路径下，所有传入的超参数会序列化存储在/opt/ml/input/config/hyperparameters.json文件中。你自己写的入口脚本只需要读取对应路径下的目标训练脚本，执行调用即可，完全可以自主控制执行逻辑，不会被默认的train文件规则限制。
• 补充说明：如果既没有配置自定义ENTRYPOINT，也没有适配上述启动逻辑，SageMaker才会去系统PATH下查找名为train的可执行文件运行，这个兜底逻辑完全可以通过修改镜像启动配置、适配参数解析规则覆盖，不存在平台层面的功能锁死。

内容的提问来源于stack exchange，提问作者terrygryffindor