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 1ea0be1b17
Improve tracing of window attn models with simpler reshape logic
2 years ago
..
_pruned
layers
__init__.py
_builder.py
_efficientnet_blocks.py
_efficientnet_builder.py
_factory.py
_features.py
_features_fx.py
_helpers.py
_hub.py
_manipulate.py
_pretrained.py
_prune.py
_registry.py
beit.py
byoanet.py
byobnet.py
cait.py
coat.py
convit.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
convmixer.py
convnext.py
crossvit.py
cspnet.py
davit.py Improve tracing of window attn models with simpler reshape logic 2 years ago
deit.py
densenet.py
dla.py
dpn.py
edgenext.py
efficientformer.py
efficientformer_v2.py
efficientnet.py
factory.py Add a deprecation phase to module re-org 2 years ago
features.py
fx_features.py
gcvit.py Improve tracing of window attn models with simpler reshape logic 2 years ago
ghostnet.py
gluon_resnet.py
gluon_xception.py
hardcorenas.py
helpers.py
hrnet.py
hub.py
inception_resnet_v2.py
inception_v3.py
inception_v4.py
levit.py
maxxvit.py
mlp_mixer.py
mobilenetv3.py
mobilevit.py
mvitv2.py
nasnet.py
nest.py
nfnet.py
pit.py
pnasnet.py
poolformer.py
pvt_v2.py
registry.py
regnet.py
res2net.py
resnest.py
resnet.py
resnetv2.py
rexnet.py
selecsls.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
senet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
sequencer.py
sknet.py
swin_transformer.py Improve tracing of window attn models with simpler reshape logic 2 years ago
swin_transformer_v2.py Improve tracing of window attn models with simpler reshape logic 2 years ago
swin_transformer_v2_cr.py Improve tracing of window attn models with simpler reshape logic 2 years ago
tnt.py
tresnet.py
twins.py
vgg.py
visformer.py
vision_transformer.py
vision_transformer_hybrid.py
vision_transformer_relpos.py
volo.py
vovnet.py
xception.py
xception_aligned.py Update ClassifierHead module, add reset() method, update in_chs -> in_features for consistency 2 years ago
xcit.py