pytorch-image-models/timm/models
Ross Wightman dc18cda2e7 efficientvit (mit) msa attention q/k/v ops need to be in float32 to train w/o NaN 2023-08-20 11:49:36 -07:00
..
_pruned
layers
__init__.py Combine ghostnetv2 with ghostnet, reduec redundancy, add weights to hf hub. 2023-08-19 23:33:43 -07:00
_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
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
efficientvit_mit.py efficientvit (mit) msa attention q/k/v ops need to be in float32 to train w/o NaN 2023-08-20 11:49:36 -07:00
efficientvit_msra.py
eva.py
factory.py
features.py
focalnet.py
fx_features.py
gcvit.py
ghostnet.py Update attrib comment to include v2 2023-08-19 23:39:09 -07:00
hardcorenas.py
helpers.py
hrnet.py
hub.py
inception_resnet_v2.py
inception_v3.py
inception_v4.py
levit.py
maxxvit.py
metaformer.py
mlp_mixer.py
mobilenetv3.py
mobilevit.py
mvitv2.py
nasnet.py
nest.py
nfnet.py
pit.py
pnasnet.py
pvt_v2.py
registry.py
regnet.py
repghost.py Weights on hf hub, bicubic yields slightly better eval 2023-08-19 16:25:45 -07:00
repvit.py
res2net.py
resnest.py
resnet.py
resnetv2.py
rexnet.py
selecsls.py
senet.py
sequencer.py
sknet.py
swin_transformer.py
swin_transformer_v2.py
swin_transformer_v2_cr.py
tnt.py
tresnet.py
twins.py
vgg.py
visformer.py
vision_transformer.py
vision_transformer_hybrid.py
vision_transformer_relpos.py
vision_transformer_sam.py
volo.py
vovnet.py
xception.py
xception_aligned.py
xcit.py