create_model.py

client.create_model(
    ModelName=“Your model name”,
    PrimaryContainer={
    “Image”: <AWS_DEEP_LEARNING_CONTAINER_URI>,
    “ModelDataUrl”: <MODEL_S3_LOCATION>,
    “Environment”: {
        “SAGEMAKER_PROGRAM”: “inference.py”,
        “SAGEMAKER_SUBMIT_DIRECTORY”: “<INFERENCE_SCRIPT_S3_LOCATION>”,
        “SAGEMAKER_CONTAINER_LOG_LEVEL”: “20”,
        “SAGEMAKER_REGION”: <REGION>,
        “TF_ENABLE_ONEDNN_OPTS”: “1”,
        “DNNL_DEFAULT_FPMATH_MODE”: “BF16”
         }
     },
     ExecutionRoleArn=‘ARN for AmazonSageMaker-ExecutionRole’
)