We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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 。提供一个接口,用户将调用这个接口侵入式地修改组网代码,标记需要监控的 Tensor 。这个接口与其他接口是互相独立的。
Tensor
该接口收到一个 Tensor 后,应当保存这个 Tensor 的信息(到某个全局变量/闭包/或者其他机制),并注册反向 hook 。在一个step 后提供某种机制,将这个 step 中记录的 Tensor 信息保存下来。
hook
step
具体的做法可以参考仓库中 report 文件夹下的 report 与 tensor_hook 之间的互动。最后,还需要提供一个 dump 记录信息的接口。
report
tensor_hoo
dump
The text was updated successfully, but these errors were encountered:
No branches or pull requests
一、问题描述 📚
在模型精度检查时,有时只需要监控特定的几个关键
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: