Skip to content

Instantly share code, notes, and snippets.

@shashankprasanna

shashankprasanna/debug_hook.py

Last active April 2, 2020 21:02
Show Gist options
  • Save shashankprasanna/924f6bde55f38e4fda662de73076c1ea to your computer and use it in GitHub Desktop.
Save shashankprasanna/924f6bde55f38e4fda662de73076c1ea to your computer and use it in GitHub Desktop.
Download ZIP
Raw
debug_hook.py
from sagemaker.debugger import Rule, DebuggerHookConfig
debugger_hook_config = DebuggerHookConfig(
hook_parameters={"save_interval": '100'},
collection_configs=[
CollectionConfig("losses"),
CollectionConfig("weights"),
CollectionConfig("gradients"),
CollectionConfig("biases")]
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment