Skip to content

Instantly share code, notes, and snippets.

@siakon89
Last active July 5, 2020 09:03
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
Star You must be signed in to star a gist
Save siakon89/1f18f11c58f9c25d326d22951a993449 to your computer and use it in GitHub Desktop.
from sagemaker.debugger import CollectionConfig, DebuggerHookConfig
bucket = sess.default_bucket()
collection_config_biases = CollectionConfig(name='biases')
collection_config_weights = CollectionConfig(name='weights')
collection_config_metrics = CollectionConfig(name='metrics')
debugger_hook_config = DebuggerHookConfig(
s3_output_path=f"s3://{bucket}/fashion-mnist/hook",
hook_parameters={
'save_interval': '100'
},
collection_configs=[
collection_config_biases,
collection_config_weights,
collection_config_metrics
]
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment