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 bb50ac4708
Add DeiT distilled weights and distilled model def. Remove some redudant ViT model args.
4 years ago
..
layers ResNetV2 pre-act and non-preact model, w/ BiT pretrained weights and support for ViT R50 model. Tweaks for in21k num_classes passing. More to do... tests failing. 4 years ago
pruned
__init__.py ResNetV2 pre-act and non-preact model, w/ BiT pretrained weights and support for ViT R50 model. Tweaks for in21k num_classes passing. More to do... tests failing. 4 years ago
cspnet.py
densenet.py
dla.py
dpn.py Remove some old __future__ imports 4 years ago
efficientnet.py
efficientnet_blocks.py
efficientnet_builder.py
factory.py ResNetV2 pre-act and non-preact model, w/ BiT pretrained weights and support for ViT R50 model. Tweaks for in21k num_classes passing. More to do... tests failing. 4 years ago
features.py
gluon_resnet.py
gluon_xception.py
helpers.py More dataset work including factories and a tensorflow datasets (TFDS) wrapper 4 years ago
hrnet.py
inception_resnet_v2.py
inception_v3.py
inception_v4.py
mobilenetv3.py
nasnet.py ResNetV2 pre-act and non-preact model, w/ BiT pretrained weights and support for ViT R50 model. Tweaks for in21k num_classes passing. More to do... tests failing. 4 years ago
pnasnet.py
registry.py
regnet.py
res2net.py
resnest.py
resnet.py Convert samples/targets in ParserImageInTar to numpy arrays, slightly less mem usage for massive datasets. Add a few more se/eca model defs to resnet.py 4 years ago
resnetv2.py Finishing adding stochastic depth support to BiT ResNetV2 models 4 years ago
rexnet.py
selecsls.py
senet.py
sknet.py
tresnet.py
vision_transformer.py Add DeiT distilled weights and distilled model def. Remove some redudant ViT model args. 4 years ago
vovnet.py
xception.py
xception_aligned.py