Skip to content

Instantly share code, notes, and snippets.

@EniasCailliau

EniasCailliau/tf_sagemaker_create_estimator.py

Created August 23, 2018 13:07
Show Gist options
  • Save EniasCailliau/257b5e86ec24518973829d6808bd558a to your computer and use it in GitHub Desktop.
Save EniasCailliau/257b5e86ec24518973829d6808bd558a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
tf_sagemaker_create_estimator.py
estimator = TensorFlow(entry_point='cnn_fashion_mnist.py',
role=role,
input_mode='Pipe',
training_steps=20000,
evaluation_steps=100,
train_instance_count=1,
train_instance_type='ml.c5.2xlarge',
base_job_name='radix_mnist_fashion')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment