Code

Add ability to get attention maps#43

Code

Merged

matsen merged 4 commits intomainmatsengrp/netam:mainfrom 42-attn-mapmatsengrp/netam:42-attn-mapCopy head branch name to clipboardJun 25, 2024

+79

Commits

Commits on Jun 22, 2024

• first attempt for attention map

  matsen

  committed
  c9068bb

  Copy full SHA for c9068bb

• working right

  matsen

  committed
  7968c5c

  Copy full SHA for 7968c5c

Commits on Jun 23, 2024

• Add axis parameter

  matsen

  committed
  13b9ceb

  Copy full SHA for 13b9ceb

• better interface

  matsen

  committed
  7452ce0

  Copy full SHA for 7452ce0