| 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