..
layers
Make a few more layers symbolically traceable (remove from FX leaf modules)
2022-03-24 21:43:56 -07:00
pruned
added efficientnet pruned weights
2020-05-03 09:29:45 +03:00
__init__.py
Merge remote-tracking branch 'origin/master' into norm_norm_norm
2022-03-21 13:41:43 -07:00
beit.py
migrate azure blob for beit checkpoints
2022-04-23 13:02:29 +08:00
byoanet.py
Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.
2022-01-25 21:54:13 -08:00
byobnet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
cait.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
coat.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
convit.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
convmixer.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
convnext.py
fix convnext args
2022-04-05 20:00:57 +08:00
crossvit.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
cspnet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
deit.py
Fix pit implementation to be clsoer to deit/levit re distillation head handling
2022-03-21 11:12:14 -07:00
densenet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
dla.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
dpn.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
efficientnet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
efficientnet_blocks.py
Merge remote-tracking branch 'origin/master' into norm_norm_norm
2022-01-25 22:20:45 -08:00
efficientnet_builder.py
Merge remote-tracking branch 'origin/master' into norm_norm_norm
2022-01-25 22:20:45 -08:00
factory.py
Revert ml-decoder changes to model factory and train script
2022-03-21 12:29:02 -07:00
features.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
fx_features.py
Make a few more layers symbolically traceable (remove from FX leaf modules)
2022-03-24 21:43:56 -07:00
ghostnet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
gluon_resnet.py
Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.
2022-01-25 21:54:13 -08:00
gluon_xception.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
hardcorenas.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
helpers.py
Add initial swinv2 tiny / small weights
2022-04-03 15:22:55 -07:00
hrnet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
hub.py
Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.
2022-01-25 21:54:13 -08:00
inception_resnet_v2.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
inception_v3.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
inception_v4.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
levit.py
Fix pit implementation to be clsoer to deit/levit re distillation head handling
2022-03-21 11:12:14 -07:00
mlp_mixer.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
mobilenetv3.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
mobilevit.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
nasnet.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
nest.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
nfnet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
pit.py
Fix pit implementation to be clsoer to deit/levit re distillation head handling
2022-03-21 11:12:14 -07:00
pnasnet.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
poolformer.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
registry.py
Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.
2022-01-25 21:54:13 -08:00
regnet.py
Update test crop for new RegNet-V weights to match Y
2022-03-23 21:40:53 -07:00
res2net.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
resnest.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
resnet.py
Missed first_conv in latest seresnext 'D' default_cfgs
2022-04-22 20:55:52 -07:00
resnetv2.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
rexnet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
selecsls.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
senet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
sknet.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
swin_transformer.py
Add initial swinv2 tiny / small weights
2022-04-03 15:22:55 -07:00
swin_transformer_v2_cr.py
Add initial swinv2 tiny / small weights
2022-04-03 15:22:55 -07:00
tnt.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
tresnet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
twins.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
vgg.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
visformer.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
vision_transformer.py
Add layer scale and parallel blocks to vision_transformer
2022-03-23 16:09:07 -07:00
vision_transformer_hybrid.py
Merge remote-tracking branch 'origin/master' into norm_norm_norm
2022-03-21 13:41:43 -07:00
volo.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
vovnet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
xception.py
Significant model refactor and additions:
2022-02-28 13:56:23 -08:00
xception_aligned.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
xcit.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
2022-03-19 14:58:54 -07:00
09e9f3defb