-
Notifications
You must be signed in to change notification settings - Fork 16
Open
Description
一、问题描述 📚
在模型精度检查时,有时只需要监控特定的几个关键 Tensor 。提供一个接口,用户将调用这个接口侵入式地修改组网代码,标记需要监控的 Tensor 。这个接口与其他接口是互相独立的。
二、任务目标 🚀
该接口收到一个 Tensor 后,应当保存这个 Tensor 的信息(到某个全局变量/闭包/或者其他机制),并注册反向 hook 。在一个step 后提供某种机制,将这个 step 中记录的 Tensor 信息保存下来。
三、 TIPS
具体的做法可以参考仓库中 report 文件夹下的 report 与 tensor_hook 之间的互动。最后,还需要提供一个 dump 记录信息的接口。
Metadata
Metadata
Assignees
Labels
No labels