fynance.models.rolling.RollMultiLayerPerceptron.register_forward_hook

RollMultiLayerPerceptron.register_forward_hook(hook)

Registers a forward hook on the module.

The hook will be called every time after forward has computed an output. It should have the following signature:

hook(module, input, output) -> None

The hook should not modify the input or output.

Returns:

torch.utils.hooks.RemovableHandle:

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