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