2025年10月30日
hook 工具随笔
摘要： hook 是 pytorch 中的一个工具，主要作用是 在模型前向/反向传播过程之前/之后执行一些自定义操作 比如说： 打印查看一些模型参数；修改梯度等等 基本形态 基本的形式是： # 自定义操作 def forward_hook(module, input, output): print(f"Mo 阅读全文
posted @ 2025-10-30 19:43 亦可九天揽月 阅读(7) 评论(0) 推荐(0)