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