jessevig / bertviz

BertViz: Visualize Attention in NLP Models (BERT, GPT2, BART, etc.)

https://towardsdatascience.com/deconstructing-bert-part-2-visualizing-the-inner-workings-of-attention-60a16d86b5c1

support BART?

LiyingCheng95 opened this issue 3 years ago · comments

Cheng Liying 程丽颖 commented 3 years ago

Hi, I'm wondering if there's a visualization tool for BART. Thanks in advance!

Jesse Vig commented 3 years ago

Hi, I haven't tried BART specifically but BertViz does support encoder-decoder models like BART: https://github.com/jessevig/bertviz#encoder-decoder-models-marianmt-etc

Please let me know if that works for you. Thanks!

Jesse Vig commented 3 years ago

Hi, I've added some sample notebooks for BART, e.g. : https://github.com/jessevig/bertviz/blob/master/notebooks/model_view_bart.ipynb