-
Notifications
You must be signed in to change notification settings - Fork 13
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
Comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
一、问题描述 📚
在模型精度检查时,有时只需要监控特定的几个关键
Tensor
。提供一个接口,用户将调用这个接口侵入式地修改组网代码,标记需要监控的Tensor
。这个接口与其他接口是互相独立的。二、任务目标 🚀
该接口收到一个
Tensor
后,应当保存这个Tensor
的信息(到某个全局变量/闭包/或者其他机制),并注册反向hook
。在一个step
后提供某种机制,将这个step
中记录的Tensor
信息保存下来。三、 TIPS
具体的做法可以参考仓库中
report
文件夹下的report
与tensor_hoo
k 之间的互动。最后,还需要提供一个dump
记录信息的接口。The text was updated successfully, but these errors were encountered: