Add TimmBackbone model #22619
amyeroberts
merged
33 commits into
huggingface:main
from
amyeroberts:add-timm-backbone-model
amyeroberts
commented
on 2023-05-11
Add test_backbone for convnext
5025fe75
Add TimmBackbone model
f20ea2d4
Add check for backbone type
1e74786c
Tidying up - config checks
3b07f877
Update convnextv2
48fcebb3
Tidy up
052531b5
Fix indices & clearer comment
35b930d2
Exceptions for config checks
b67dc1dc
Correclty update config for tests
f4c0e999
Safer imports
a77fdf81
Safer safer imports
197964a8
Fix where decorators go
b2454f99
Update import logic and backbone tests
e71549b8
More import fixes
7bd302d2
Fixup
15a1e2b4
Only import all_models if torch available
2144ba56
Fix kwarg updates in from_pretrained & main rebase
a40847f8
Tidy up
bec471b1
Add tests for AutoBackbone
4a648cf5
Tidy up
f27271cb
Fix import error
ba819e77
Fix up
0f56ef46
Install nattan in doc_test_job
5b5d8f42
Revert back to setting self._out_xxx directly
146d7658
Bug fix - out_indices mapping from out_features
b1547acf
Fix tests
22cb52bb
Dont accept output_loading_info for Timm models
b9917416
Set out_xxx and don't remap
d47ac736
Use smaller checkpoint for test
bdb6a2fc
Don't remap timm indices - check out_indices based on stage names
63f61654
Skip test as it's n/a
8316f23b
Apply suggestions from code review
c0b3ef8a
Cleaner imports / spelling is hard
1748aa25
Login to write a write a comment.
Login via GitHub