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.

Sign up for GitHub

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

Jump to bottom

关于visualizer的嵌入问题 #4

Open
Lsz-20 opened this issue Nov 22, 2021 · 2 comments
Open

关于visualizer的嵌入问题 #4

Lsz-20 opened this issue Nov 22, 2021 · 2 comments

Comments

@Lsz-20
Copy link

Lsz-20 commented Nov 22, 2021

感谢您的代码，有些问题想要请教一下
目前我已经在框架下搭好了我的网络模型，所以只想要把attention可视化的这部分嵌入到网络中，看到您的文件中涉及的timm（已有）和visualizer部分，是否可以只使用您visualizer下的内容完成操作呢？
感谢回复
@luo3300612
Copy link
Owner

luo3300612 commented Dec 6, 2021 •
edited
Loading

你好，可以的，我文件中的timm只是用来举例子的

@Lsz-20
Copy link
Author

Lsz-20 commented Dec 6, 2021

你好，可以的，我文件中的timm只是用来举例子的

感谢您的回复 参考您的操作，我在我的model的qkv attention部分加入了该操作，但是最终没有图像生成（我检查了11 12 13中的attention_maps的变量是正常获取到的）或许您有什么建议吗？

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@luo3300612 @Lsz-20