Skip to content

Instantly share code, notes, and snippets.

@siakon89

siakon89/custom_debugger_hook.py

Created July 5, 2020 09:16
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save siakon89/167a4eb4ee6853c5461e7fcc3d57cf81 to your computer and use it in GitHub Desktop.
Save siakon89/167a4eb4ee6853c5461e7fcc3d57cf81 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
custom_debugger_hook.py
debugger_hook_config = DebuggerHookConfig(
s3_output_path=f"s3://{bucket}/fashion-mnist/hook",
collection_configs=[
CollectionConfig(
name="all_tensors",
parameters={
"include_regex": ".*",
"save_steps": "1, 2, 3"
}
)
]
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment