Skip to content

Instantly share code, notes, and snippets.

Created April 17, 2019 02:39
What would you like to do?
# Simply pass in a learner and the module you would like to instrument
def probeModule(learn, module):
hook = StoreHook(learn, modules=flatten_model(module))
learn.callbacks += [ hook ]
return hook
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment