yojayc

What is the meaning of 'flat' variable in vit_rollout.py?

1

The code doesn't use the variable after reevaluating it in line 27 https://github.com/jacobgil/vit-explain/blob/15a81d355a5aa6128ea4e71bbd56c28888d0f33b/vit_rollout.py#L27

Are there any differences between nn.TransformerEncoder and class Block in transformer_model.py? Have you ever tried to use class Block instead of nn.TransformerEncoder in aggregate module just like what you do...