We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ef5930c commit d2fb9dbCopy full SHA for d2fb9db
google/cloud/aiplatform/training_jobs.py
@@ -2406,9 +2406,9 @@ def __init__(
2406
):
2407
"""Constructs a Custom Container Training Job.
2408
2409
- job = aiplatform.CustomTrainingJob(
+ job = aiplatform.CustomContainerTrainingJob(
2410
display_name='test-train',
2411
- container_uri='gcr.io/cloud-aiplatform/training/tf-cpu.2-2:latest',
+ container_uri='gcr.io/my_project_id/my_image_name:tag',
2412
command=['python3', 'run_script.py']
2413
model_serving_container_image_uri='gcr.io/my-trainer/serving:1',
2414
model_serving_container_predict_route='predict',
0 commit comments