Skip to content

Instantly share code, notes, and snippets.

@shashankprasanna

shashankprasanna/sagemaker-spot-simulate.py

Created April 24, 2020 22:23
Show Gist options
  • Save shashankprasanna/6388bb12ddddf1b91b17c822ebb8f54d to your computer and use it in GitHub Desktop.
Save shashankprasanna/6388bb12ddddf1b91b17c822ebb8f54d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
sagemaker-spot-simulate.py
tf_estimator = TensorFlow(entry_point = 'cifar10-training-sagemaker.py',
...
...
checkpoint_s3_uri = tf_estimator.checkpoint_s3_uri,
train_use_spot_instances = True,
train_max_wait = 7200)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment