自定义入口点 - AWS 深度学习容器

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

自定义入口点

对于某些图像,Deep Learning Containers 使用自定义入口点脚本。如果您要使用自己的入口点,可以按如下方式覆盖入口点。

修改包含任务定义JSON的文件中的entryPoint参数。包括自定义入口点脚本的文件路径。此处显示了一个示例。

"entryPoint":[ "sh", "-c", "/usr/local/bin/multi-model-server --start --foreground --mms-config /home/model-server/config.properties --models densenet=https://dlc-samples.s3.amazonaws.com/pytorch/multi-model-server/densenet/densenet.mar"],