Error

[myt889]

您好，感谢您的代码，有几个问题想要请教您：
目的：可视化Query2Label这个多标签分类模型的decoder中的multi-head attention.

1. 首先Query2Label模型是由CNN backbone+ Transformer decoder组成的，其中decoder部分由torch.nn .MultiheadAttention()构成，比如有4个heads。CNN backbone，Transformer网络分别在单独的.py文件搭建。
   目前我已经在Transformer.py文件中按您的建议步骤，在指定位置输入代码：
   from visualizer import get_local
   @get_local("attention_map")
   并且在需要网络输出attention的def forward()前，输入代码：
   @get_local('attention_map')
   def forward(self, tgt,....):
   ..........
   attention_map = out
   return out
   2. 按您的指南，写了对应的可视化代码。
   3. 运行代码，出现如下错误：
   File "/home/tym/myprojects/Dection/visualizer/visualizer.py", line 15, in call
   c = Bytecode.from_code(func.code)
   AttributeError: type object 'Transformer' has no attribute 'code'

我想问您，（1）由于模型复杂，需要多个模块嵌套，那上述步骤1，对么？（2）运行代码错误的原因是什么？如何解决，谢谢您。