Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

标记tensor功能 #78

Open
feifei-111 opened this issue Jun 20, 2023 · 0 comments
Open

标记tensor功能 #78

feifei-111 opened this issue Jun 20, 2023 · 0 comments

Comments

@feifei-111
Copy link
Collaborator

feifei-111 commented Jun 20, 2023

一、问题描述 📚

在模型精度检查时,有时只需要监控特定的几个关键 Tensor 。提供一个接口,用户将调用这个接口侵入式地修改组网代码,标记需要监控的 Tensor 。这个接口与其他接口是互相独立的。

二、任务目标 🚀

该接口收到一个 Tensor 后,应当保存这个 Tensor 的信息(到某个全局变量/闭包/或者其他机制),并注册反向 hook 。在一个step 后提供某种机制,将这个 step 中记录的 Tensor 信息保存下来。

三、 TIPS

具体的做法可以参考仓库中 report 文件夹下的 reporttensor_hook 之间的互动。最后,还需要提供一个 dump 记录信息的接口。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant