Please refer to the example notebooks in Amazon SageMaker Examples repository
The above notebooks come with example scripts which can be used through SageMaker. Some more example scripts are here in scripts/
Example configurations for saving tensors through SageMaker Python SDK
Example configurations for saving tensors through the hook are available at docs/sagemaker.md
Example configurations for running rules through SageMaker Python SDK
Example configurations for saving tensors through the hook are available at docs/sagemaker.md
from smdebug.rules import invoke_rule
from smdebug.trials import create_trial
trial = create_trial('s3://bucket/prefix')
rule_obj = CustomRule(trial, param=value)
invoke_rule(rule_obj, start_step=0, end_step=10)