SuperSupermoon / MedViLL

MedViLL official code. (Published IEEE JBHI 2021)
MIT License
83 stars 11 forks source link

Attention Visualization #13

Closed khoangothe closed 1 year ago

khoangothe commented 1 year ago

Hi! Thanks for the awesome work. Could you share the code for the attention visualization experiment at the end of the paper? Or could you share which attention layers were used for the visualization. Thank you!

Best, Khoa

SuperSupermoon commented 1 year ago

Thanks! I will update visualize code soon! Please stay tuned!