PyTorch Hooks Explained - In-depth Tutorial

Publié le: 25 août 2020
sur la chaîne: Elliot Waite

35,415

1.4k

UPDATE: `register_backward_hook()` has been deprecated in favor of `register_full_backward_hook()`. You can read more about `register_full_backward_hook()` here:
https://pytorch.org/docs/stable/gener...

In this video, I explain how hooks work in PyTorch. Hooks allow you to inject code into different parts of the computational graphs (in both the forward graph and the backward graph). The four types of hooks that are available in PyTorch are:
Tensor.register_hook (https://pytorch.org/docs/stable/gener...)
Module.register_forward_pre_hook (https://pytorch.org/docs/master/gener...)
Module.register_forward_hook (https://pytorch.org/docs/master/gener...)
Module.register_backward_hook (https://pytorch.org/docs/master/gener...)

My previous video, "PyTorch Autograd Explained - In-depth Tutorial":
📼    • PyTorch Autograd Explained - In-depth...

GitHub code examples of using hooks:
💻 https://github.com/elliotwaite/pytorc...

Join our Discord community:
💬   / discord

Connect with me:
🐦 Twitter -   / elliotwaite
📷 Instagram -   / elliotwaite
👱 Facebook -   / elliotwaite
💼 LinkedIn -   / elliotwaite

🎵 Kazukii - Return
→   / ohthatkazuki
→ https://open.spotify.com/artist/5d07M...
→    / officialkazuki

Sur cette page du site, vous pouvez voir la vidéo en ligne PyTorch Hooks Explained - In-depth Tutorial durée heure minute seconde en bonne qualité , qui a été Téléchargé par l'utilisateur Elliot Waite 25 août 2020, Partagez le lien avec vos amis et connaissances, sur youtube cette vidéo a déjà été regardée 35,415 fois et il a aimé 1.4 mille téléspectateurs. Bon visionnage!

9,580