public AlgorithmSpecification withTrainingImage(String trainingImage) {
        setTrainingImage(trainingImage);
        return this;
    }
<p>
The registry path of the Docker image that contains the training algorithm. For information about docker registry
paths for SageMaker built-in algorithms, see <a
href="https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html">Docker Registry
Paths and Example Code</a> in the <i>Amazon SageMaker developer guide</i>. SageMaker supports both
<code>registry/repository[:tag]</code> and <code>registry/repository[@digest]</code> image path formats. For more
information about using your custom training container, see <a
href="https://docs.aws.amazon.com/sagemaker/latest/dg/your-algorithms.html">Using Your Own Algorithms with Amazon
SageMaker</a>.
</p>
<note>
<p>
You must specify either the algorithm name to the <code>AlgorithmName</code> parameter or the image URI of the
algorithm container to the <code>TrainingImage</code> parameter.
</p>
<p>
For more information, see the note in the <code>AlgorithmName</code> parameter description.
</p>
</note>

@param trainingImage
       The registry path of the Docker image that contains the training algorithm. For information about docker
       registry paths for SageMaker built-in algorithms, see <a
       href="https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html">Docker
       Registry Paths and Example Code</a> in the <i>Amazon SageMaker developer guide</i>. SageMaker supports
       both <code>registry/repository[:tag]</code> and <code>registry/repository[@digest]</code> image path
       formats. For more information about using your custom training container, see <a
       href="https://docs.aws.amazon.com/sagemaker/latest/dg/your-algorithms.html">Using Your Own Algorithms with
       Amazon SageMaker</a>.</p> <note>
       <p>
       You must specify either the algorithm name to the <code>AlgorithmName</code> parameter or the image URI of
       the algorithm container to the <code>TrainingImage</code> parameter.
       </p>
       <p>
       For more information, see the note in the <code>AlgorithmName</code> parameter description.
       </p>
@return Returns a reference to this object so that method calls can be chained together.
aws-java-sdk-sagemaker/src/main/java/com/amazonaws/services/sagemaker/model/AlgorithmSpecification.java