Pytorch Module Hook

Creator

Creator

Seonglae Cho

Created

Created

2024 Apr 14 6:4

Editor

Editor

Seonglae Cho

Edited

Edited

2024 Apr 15 12:35

Refs

Refs

register_forward_hook()

register_full_backward_hook()

register_forward_pre_hook()

Modules — PyTorch 2.2 documentation

Building blocks of stateful computation. PyTorch provides a robust library of modules and makes it simple to define new custom modules, allowing for easy construction of elaborate, multi-layer neural networks.

Modules — PyTorch 2.2 documentation

https://pytorch.org/docs/stable/notes/modules.html#module-hooks

torch.nn.modules.module.register_module_forward_hook — PyTorch 2.2 documentation

This adds global state to the nn.module module and it is only intended for debugging/profiling purposes.

torch.nn.modules.module.register_module_forward_hook — PyTorch 2.2 documentation

https://pytorch.org/docs/stable/generated/torch.nn.modules.module.register_module_forward_hook.html

Recommendations

//////////