开发者

pytorch中的hook机制

0
  • pytorch中的hook机制register_forward_hook

    目录1、hook背景2、源码阅读3、定义一个用于测试hooker的类4、定义hook函数5、对需要的层注册hook6、测试forward()返回的特征和hook记录的是否一致6.1测试forward()提供的输入输出特征6.2hook记录的输入特征和输[详细]

    2022-12-12 13:04 分类:开发