pyvene.models.modeling_utils.remove_forward_hooks

pyvene.models.modeling_utils.remove_forward_hooks#

remove_forward_hooks(main_module: Module)[source]#

Function to remove all forward and pre-forward hooks from a module and

its sub-modules.