Skip to content

Instantly share code, notes, and snippets.

@siakon89
Created July 5, 2020 10:49
Show Gist options
  • Save siakon89/def472fac41acc2ae5ec923a3f50f86d to your computer and use it in GitHub Desktop.
Save siakon89/def472fac41acc2ae5ec923a3f50f86d to your computer and use it in GitHub Desktop.
from sagemaker.debugger import Rule, rule_configs
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,
debugger_hook_config=debugger_hook_config,
rules=[
Rule.sagemaker(rule_configs.overfit()),
Rule.sagemaker(rule_configs.loss_not_decreasing())
],
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment