Skip to content

Instantly share code, notes, and snippets.

@icoxfog417
Created April 8, 2019 06:13
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save icoxfog417/a97713f4b5398388547accca4775f075 to your computer and use it in GitHub Desktop.
Save icoxfog417/a97713f4b5398388547accca4775f075 to your computer and use it in GitHub Desktop.
allennlp_tutorial_estimator.py
estimator = PyTorch(entry_point="tuning.py",
role=role,
framework_version="1.0.0",
train_instance_count=1,
train_instance_type="ml.p2.8xlarge",
hyperparameters={
"train-file-name": os.path.basename(s3_paths[0]),
"validation": os.path.basename(s3_paths[1]),
"epochs": 10
})
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment