pytorch-image-models/timm/models
Ross Wightman 7b6a406474 remove swin debug prints 2024-07-24 21:05:56 -07:00
..
_pruned
layers
__init__.py
_builder.py
_efficientnet_blocks.py
_efficientnet_builder.py
_factory.py
_features.py
_features_fx.py
_helpers.py _orig_mod still causing issues even though I thought it was fixed in pytorch, add unwrap / clean helpers 2024-07-19 11:03:45 -07:00
_hub.py
_manipulate.py
_pretrained.py
_prune.py
_registry.py
beit.py Add 'qkv_bias_separate' flag for EVA/beit/swinv2 attn modules to allow an override for easy quantization wrappers. Fix #2098 2024-07-08 13:48:38 -07:00
byoanet.py
byobnet.py Add weights for test models 2024-07-22 10:08:57 -07:00
cait.py
coat.py
convit.py set_input_size initial impl for vit & swin v1. Move HybridEmbed to own location in timm/layers 2024-07-17 15:25:48 -07:00
convmixer.py
convnext.py
crossvit.py
cspnet.py
davit.py
deit.py
densenet.py
dla.py
dpn.py
edgenext.py
efficientformer.py
efficientformer_v2.py
efficientnet.py Add weights for test models 2024-07-22 10:08:57 -07:00
efficientvit_mit.py
efficientvit_msra.py
eva.py Add 'qkv_bias_separate' flag for EVA/beit/swinv2 attn modules to allow an override for easy quantization wrappers. Fix #2098 2024-07-08 13:48:38 -07:00
factory.py
fastvit.py
features.py
focalnet.py
fx_features.py
gcvit.py
ghostnet.py
hardcorenas.py
helpers.py
hgnet.py
hiera.py Add to 'abswin' hiera models for train trials 2024-07-19 11:05:31 -07:00
hrnet.py Fix typo in type annotations in timm.models.hrnet 2024-07-08 00:53:16 +02:00
hub.py
inception_next.py
inception_resnet_v2.py
inception_v3.py
inception_v4.py
levit.py
maxxvit.py
metaformer.py
mlp_mixer.py
mobilenetv3.py Add conv_large mobilenetv3 aa/blur model defs 2024-07-08 13:50:05 -07:00
mobilevit.py
mvitv2.py
nasnet.py
nest.py
nextvit.py
nfnet.py
pit.py
pnasnet.py
pvt_v2.py
registry.py
regnet.py
repghost.py
repvit.py
res2net.py
resnest.py
resnet.py Make channels for classic resnet configurable 2024-07-22 10:47:40 -07:00
resnetv2.py
rexnet.py
selecsls.py
senet.py
sequencer.py
sknet.py
swin_transformer.py set_input_size(), always_partition, strict_img_size, dynamic mask option for all swin models. More flexibility in resolution, window resizing. 2024-07-24 16:41:31 -07:00
swin_transformer_v2.py remove swin debug prints 2024-07-24 21:05:56 -07:00
swin_transformer_v2_cr.py remove swin debug prints 2024-07-24 21:05:56 -07:00
tiny_vit.py
tnt.py
tresnet.py
twins.py
vgg.py
visformer.py
vision_transformer.py Add weights for test models 2024-07-22 10:08:57 -07:00
vision_transformer_hybrid.py set_input_size initial impl for vit & swin v1. Move HybridEmbed to own location in timm/layers 2024-07-17 15:25:48 -07:00
vision_transformer_relpos.py
vision_transformer_sam.py
vitamin.py set_input_size initial impl for vit & swin v1. Move HybridEmbed to own location in timm/layers 2024-07-17 15:25:48 -07:00
volo.py
vovnet.py
xception.py
xception_aligned.py
xcit.py