技术栈

hook钩子函数使用

qq_41920323
1 年前
pytorch·hook钩子函数使用
pytorch中的hook机制register_forward_hook上篇文章主要介绍了hook钩子函数的大致使用流程,本篇文章主要介绍pytorch中的hook机制register_forward_hook,手动在forward之前注册hook,hook在forward执行以后被自动执行。