search

mandarjoshi90 / coref

BERT for Coreference Resolution
Apache License 2.0
440 stars 92 forks source link

Access attention weights #72

Open prasad4fun opened 3 years ago

prasad4fun commented 3 years ago

Hi @mandarjoshi90 ,

Thanks for your work. I am trying to visualize Spanbert base attention weights. when I read through the code couldn't find any param to get them.

Could you help me getting attention weights across 12 layers?