Skip to content

标记tensor功能 #78

@feifei-111

Description

@feifei-111

一、问题描述 📚

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

二、任务目标 🚀

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

三、 TIPS

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions