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 83b40c5a58
Last batch of small model weights (for now). mobilenetv3_small 050/075/100 and updated mnasnet_small with lambc/lamb optimizer.
3 years ago
..
layers
pruned
__init__.py
beit.py
byoanet.py
byobnet.py
cait.py
coat.py
convit.py Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones. 3 years ago
convmixer.py
convnext.py
crossvit.py
cspnet.py
densenet.py
dla.py
dpn.py
efficientnet.py
efficientnet_blocks.py
efficientnet_builder.py
factory.py
features.py
fx_features.py
ghostnet.py Fix a few issues loading pretrained vit/bit npz weights w/ num_classes=0 __init__ arg. Missed a few other small classifier handling detail on Mlp, GhostNet, Levit. Should fix #713 3 years ago
gluon_resnet.py
gluon_xception.py
hardcorenas.py
helpers.py
hrnet.py
hub.py
inception_resnet_v2.py
inception_v3.py
inception_v4.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
levit.py
mlp_mixer.py
mobilenetv3.py
nasnet.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
nest.py
nfnet.py
pit.py
pnasnet.py
registry.py
regnet.py
res2net.py
resnest.py Add non-local and BAT attention. Merge attn and self-attn factories into one. Add attention references to README. Add mlp 'mode' to ECA. 4 years ago
resnet.py
resnetv2.py
rexnet.py
selecsls.py
senet.py Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there. 4 years ago
sknet.py
swin_transformer.py
tnt.py rename notrace registration and standardize trace_utils imports 3 years ago
tresnet.py
twins.py rename notrace registration and standardize trace_utils imports 3 years ago
vgg.py
visformer.py
vision_transformer.py
vision_transformer_hybrid.py
vovnet.py
xception.py
xception_aligned.py
xcit.py Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones. 3 years ago