How can plot the heatmap of the self-attentions?

lucidrains / TimeSformer-pytorch

Implementation of TimeSformer from Facebook AI, a pure attention-based solution for video classification

MIT License

689 stars 86 forks source link

Open YNawal opened 3 years ago

YNawal commented 3 years ago

Hello any help about piloting the heatmap of the self attention? Thanks @lucidrains