tensorplay.nn.modules.module.register_module_parameter_registration_hook

tensorplay.nn.modules.module.register_module_parameter_registration_hook(hook: Callable[[...], None]) → RemovableHandle[source]

Register a parameter registration hook common to all modules.

Warning

This adds global state to the nn.Module module

The hook will be called every time register_parameter() is invoked. It should have the following signature:

hook(module, name, param) -> None or new parameter

The hook can modify the input or return a single modified value in the hook.

Returns:

a handle that can be used to remove the added hook by calling handle.remove()

Return type:

tensorplay.utils.hooks.RemovableHandle