Fix samvit bug, add F.sdpa support and ROPE option (#1920) · huggingface/pytorch-image-models@3055411 · GitHub

Commit

Fix samvit bug, add F.sdpa support and ROPE option (#1920)

Browse files

* Fix a bug I introduced in samvit, add F.sdpa support and ROPE option to samvit, neck is LayerNorm if not used and standard classifier used

* Add attn dropout to F.sdpa

* Fix fx trace for sam vit

* Fixing torchscript issues in samvit

* Another torchscript fix

* samvit head fc name fix

Loading branch information

rwightman authored Aug 21, 2023

1 parent 300f54a commit 3055411