wax.modules.update_on_event

Apply a module when an event occur otherwise return last computed output.

Part of this module may be integrated in Haiku. See: https://github.com/deepmind/dm-haiku/issues/126

Functions

set_params_or_state_dict(module_name, …)	Returns module parameters or state for the given module or submodules.
set_state_from_dict(self, next_state_dict)	Set state keyed by name for this module and submodules.

Classes

UpdateOnEvent(*args, **kwargs)

Apply a module when an event occur otherwise return last computed output.