Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
model_dir = '/opt/ml/model'
train_instance_type = 'ml.c5.2xlarge'
hyperparameters = {'epochs': 10, 'batch_size': 256, 'learning_rate': 0.001}
estimator = TensorFlow(
entry_point='model.py',
train_instance_type=train_instance_type,
train_instance_count=1,
model_dir=model_dir,
hyperparameters=hyperparameters,
role=sagemaker.get_execution_role(),
base_job_name='tf-fashion-mnist',
framework_version='1.15',
py_version='py3',
script_mode=True
)
estimator.fit(inputs)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.