| # 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