Skip to content

Instantly share code, notes, and snippets.

@shashankprasanna

shashankprasanna/smdebug_rule_invoke.py

Created April 2, 2020 21:19
Show Gist options
  • Save shashankprasanna/8e9447dc7fc66f5a9e796af688622ef2 to your computer and use it in GitHub Desktop.
Save shashankprasanna/8e9447dc7fc66f5a9e796af688622ef2 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
smdebug_rule_invoke.py
from smdebug.rules import invoke_rule
from smdebug.trials import create_trial
trial = create_trial(path=’./smd_outputs/<JOB_NAME>)
rule_obj = CustomVanishingGradientRule(trial, threshold=0.0001)
invoke_rule(rule_obj, start_step=0, end_step=None)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment