fixed ticket #11 and a bug

Open yiyixuxu opened this issue 3 years ago • 0 comments

fixed the code example in README.md (issue #11) "unexpected keyword argument 'head_fusion'
reset the value for attribute self.attentions and self.attention_gradients before each forward pass so the we can use the VITAttentionGradRollout object to explain more than one images

Mar 12 '22 05:03 yiyixuxu