MSN (Masked Siamese Networks) for ViT #18815
sgugger
merged
30 commits into
huggingface:main
from
feat/msn
feat: modeling and conversion scripts for msn.
3c03aca0
chore: change license year.
26d57972
chore: remove unneeded modules.
48c3732d
feat: direct loading of state_dict from remote url.
3a965dd5
fix: import paths.
aa383d0e
add: rest of the files.
56f5be29
add and fix rest of the files.
fd9d18f4
chore: formatting.
7887834b
code quality fix.
65f557c3
chore: remove pooler.
2d933679
Merge branch 'main' into feat/msn
1bd9b21a
feat: add classification top.
e4493fa8
fix: configuration object.
92f3ab41
add: initial test cases (one failing).
afc220a2
fix: basemodeloutput.
5ba5e80e
add: caution on using the classification head.
d4e14127
add: rest of the model related files.
d49b078a
add: vit msn readme.
1db608ce
sayakpaul
commented
on 2022-09-13
sayakpaul
commented
on 2022-09-13
sayakpaul
commented
on 2022-09-13
sayakpaul
commented
on 2022-09-13
sayakpaul
commented
on 2022-09-13
fix: copied from statement.
045d71ab
Merge branch 'main' into feat/msn
de366f9d
sayakpaul
changed the title
fix: dummy objects.
2bc91ec2
add: ViTMSNPreTrainedModel to inits.
29729e51
fix: repo consistency.
98b428d6
minor change in the model doc.
5b8ecb51
fix: tests.
0d2ce07e
Empty-Commit
65725c44
Update src/transformers/models/vit_msn/configuration_vit_msn.py
4c2d732a
address PR comments.
0da81a48
Update src/transformers/models/vit_msn/modeling_vit_msn.py
b7b17cb9
chore: put model in no_grad() and formatting.
89c24cc9
Login to write a write a comment.
Login via GitHub