Skip to content

Instantly share code, notes, and snippets.

@juliensimon
Created June 2, 2018 08:03
Show Gist options
  • Save juliensimon/a93769c712b522765bc5c08a2408bb7a to your computer and use it in GitHub Desktop.
Save juliensimon/a93769c712b522765bc5c08a2408bb7a to your computer and use it in GitHub Desktop.
pytorch-estimator.py
output_path = 's3://{}/{}/output'.format(sess.default_bucket(), repo_name)
image_name = '{}.dkr.ecr.{}.amazonaws.com/{}:latest'.format(account, region, repo_name)
print(output_path)
print(image_name)
estimator = sagemaker.estimator.Estimator(
image_name=image_name,
base_job_name=base_job_name,
role=role,
train_instance_count=1,
train_instance_type=train_instance_type,
output_path=output_path,
sagemaker_session=sess)
estimator.set_hyperparameters(lr=0.01, epochs=10, batch_size=batch_size)
estimator.fit({'training': train_input_path, 'validation': validation_input_path})
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment