ml-jku / hopfield-layers

Hopfield Networks is All You Need

Home Page:https://ml-jku.github.io/hopfield-layers/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How do I get attention map?

hoangphucITJP opened this issue · comments

I'm trying to apply HopfieldPooling to the MIL problem. How do I get the attention map?