You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pytorch-image-models/timm/models
Ross Wightman e858912e0c
Add brute-force checkpoint remapping option
2 years ago
..
layers Add CL SE module 2 years ago
pruned
__init__.py
beit.py BEiT-V2 checkpoints didn't remove 'module' from weights, adapt checkpoint filter 2 years ago
byoanet.py
byobnet.py
cait.py
coat.py
convit.py
convmixer.py
convnext.py
crossvit.py
cspnet.py
deit.py
densenet.py
dla.py Fix type in dla weight update 2 years ago
dpn.py
edgenext.py
efficientformer.py
efficientnet.py Update a bunch of weights with external links to timm release assets. Fixes issue with *aliyuncs.com returning forbidden. Did pickle scan / verify and re-hash. Add TresNet-V2-L weights. 2 years ago
efficientnet_blocks.py
efficientnet_builder.py
factory.py
features.py
fx_features.py
gcvit.py Make gcvit window size ratio based to improve resolution changing support #1449. Change default init to original. 2 years ago
ghostnet.py
gluon_resnet.py
gluon_xception.py
hardcorenas.py Update a bunch of weights with external links to timm release assets. Fixes issue with *aliyuncs.com returning forbidden. Did pickle scan / verify and re-hash. Add TresNet-V2-L weights. 2 years ago
helpers.py Add brute-force checkpoint remapping option 2 years ago
hrnet.py
hub.py Fix errant import 2 years ago
inception_resnet_v2.py
inception_v3.py
inception_v4.py
levit.py
maxxvit.py Update maxvit_tiny_256 weight to better iter, add coatnet / maxvit / maxxvit model defs for future runs 2 years ago
mlp_mixer.py Update a bunch of weights with external links to timm release assets. Fixes issue with *aliyuncs.com returning forbidden. Did pickle scan / verify and re-hash. Add TresNet-V2-L weights. 2 years ago
mobilenetv3.py Update a bunch of weights with external links to timm release assets. Fixes issue with *aliyuncs.com returning forbidden. Did pickle scan / verify and re-hash. Add TresNet-V2-L weights. 2 years ago
mobilevit.py
mvitv2.py pretrained_cfg entry for mvitv2_small_cls 2 years ago
nasnet.py Update a bunch of weights with external links to timm release assets. Fixes issue with *aliyuncs.com returning forbidden. Did pickle scan / verify and re-hash. Add TresNet-V2-L weights. 2 years ago
nest.py
nfnet.py
pit.py
pnasnet.py
poolformer.py
pvt_v2.py
registry.py
regnet.py
res2net.py
resnest.py
resnet.py Update a bunch of weights with external links to timm release assets. Fixes issue with *aliyuncs.com returning forbidden. Did pickle scan / verify and re-hash. Add TresNet-V2-L weights. 2 years ago
resnetv2.py
rexnet.py
selecsls.py
senet.py Update a bunch of weights with external links to timm release assets. Fixes issue with *aliyuncs.com returning forbidden. Did pickle scan / verify and re-hash. Add TresNet-V2-L weights. 2 years ago
sequencer.py
sknet.py
swin_transformer.py Update a bunch of weights with external links to timm release assets. Fixes issue with *aliyuncs.com returning forbidden. Did pickle scan / verify and re-hash. Add TresNet-V2-L weights. 2 years ago
swin_transformer_v2.py
swin_transformer_v2_cr.py
tnt.py
tresnet.py Update tresnet features_info for v2 2 years ago
twins.py
vgg.py
visformer.py
vision_transformer.py Add hf hub entries for laion2b clip models, add huggingface_hub dependency, update some setup/reqs, torch >= 1.7 2 years ago
vision_transformer_hybrid.py Adding support for fine-tune CLIP LAION-2B image tower weights for B/32, L/14, H/14 and g/14. Still WIP 2 years ago
vision_transformer_relpos.py
volo.py
vovnet.py
xception.py
xception_aligned.py
xcit.py