Is it possible SageMaker HyperParameter Tuning Job without sagemaker-training tool kit

0

I had made my custom training image so It can be conducted through CreateTrainingJob, not sagemaker training took kit (requiring "ContainerEntrypoint" option).

But when I'm trying to run HyperParameter Tuning Job, it is not allowed to add "ContainerEntrypoint" option in "AlgorithmSpecification" field.

Is it impossible to run HyperParameter Tuning Job with training images that can not be run without sagemaker training toolkit?

Thanks!

1 Answer
0
Accepted Answer

Hi there!

For custom training image, you can specify the entrypoint in your Dockerfile.

Below are some code snippet as well as links you can use as reference:

https://github.com/aws/amazon-sagemaker-examples/blob/main/hyperparameter_tuning/rapids_bring_your_own/code/Dockerfile

ENTRYPOINT ["./entrypoint.sh"]

https://github.com/aws/amazon-sagemaker-examples/blob/main/hyperparameter_tuning/keras_bring_your_own/Dockerfile

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

Furthermore, SageMaker Training Toolkit is a nicely wrapped up python package for you to use and eases the process of creating a custom training image, it's no different from implementing the logic yourself.

So it is definitely possible to run HyperParameter Tuning Job using custom containers without using our SageMaker Training Toolkit.

BenH
answered 19 days ago
  • Thanks a lot for your examples and snippets! I will reference them right now!

You are not logged in. Log in to post an answer.

A good answer clearly answers the question and provides constructive feedback and encourages professional growth in the question asker.

Guidelines for Answering Questions