Skip to content

Instantly share code, notes, and snippets.

@siakon89

siakon89/estimator_hook.py

Created July 5, 2020 09:14
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 siakon89/80143d60b7eeaee3de92abea02a2b2ad to your computer and use it in GitHub Desktop.
Save siakon89/80143d60b7eeaee3de92abea02a2b2ad to your computer and use it in GitHub Desktop.
Download ZIP
Raw
estimator_hook.py
hyperparameters = {'epochs': 10, 'batch_size': 256, 'learning_rate': 0.001}
estimator = TensorFlow(
entry_point='model.py',
train_instance_type='ml.p3.2xlarge',
train_instance_count=1,
model_dir='/opt/ml/model',
hyperparameters=hyperparameters,
role=sagemaker.get_execution_role(),
base_job_name='tf-fashion-mnist',
framework_version='1.15',
py_version='py3',
script_mode=True,
debugger_hook_config=debugger_hook_config,
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment