Skip to content

Instantly share code, notes, and snippets.

@shashankprasanna

shashankprasanna/ai-accelerator-1.py

Created October 20, 2020 08:03
Show Gist options
  • Save shashankprasanna/d4719b4cc50784e01fba160ac6b32732 to your computer and use it in GitHub Desktop.
Save shashankprasanna/d4719b4cc50784e01fba160ac6b32732 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
ai-accelerator-1.py
from sagemaker.tensorflow.model import TensorFlowModel, TensorFlowPredictor
prefix = 'keras_models'
s3_model_path = sess.upload_data(path='model.tar.gz', key_prefix=prefix)
model = TensorFlowModel(model_data=s3_model_path,
framework_version='1.15',
role=role,
predictor_cls = TensorFlowPredictor,
sagemaker_session=sess)
predictor_gpu = model.deploy(initial_instance_count=1,
instance_type='ml.g4dn.xlarge')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment