# 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