marctemp

from sagemaker.estimator import Estimator

estimator = Estimator(image_uri='<ECR REPO URL>:latest',
                      role='<SAGEMAKER ROLE ARN>',
                      base_job_name='bert-training-job',
                      instance_count=1,
                      instance_type='ml.m5.2xlarge',
                      source_dir='/var/bert/model_pkg/',
                      entry_point='model_train.py'
                     )