| .. |
|
layers
|
Add 'fast' layer norm that doesn't cast to float32, support APEX LN impl for slight speed gain, update norm and act factories, tweak SE for ability to disable bias (needed by GCVit)
|
2 years ago |
|
pruned
|
added efficientnet pruned weights
|
4 years ago |
|
__init__.py
|
Add MViT (Multi-Scale) V2
|
2 years ago |
|
beit.py
|
Remove some redundant requires_grad=True from nn.Parameter in third party code
|
2 years ago |
|
byoanet.py
|
Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.
|
2 years ago |
|
byobnet.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
cait.py
|
Remove some redundant requires_grad=True from nn.Parameter in third party code
|
2 years ago |
|
coat.py
|
Significant model refactor and additions:
|
2 years ago |
|
convit.py
|
Significant model refactor and additions:
|
2 years ago |
|
convmixer.py
|
Significant model refactor and additions:
|
2 years ago |
|
convnext.py
|
Add timm ConvNeXt 'atto' weights, change test resolution for FB ConvNeXt 224x224 weights, add support for different dw kernel_size
|
2 years ago |
|
crossvit.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
cspnet.py
|
Add weights for two 'Edge' block (3x3->1x1) variants of CS3 networks.
|
2 years ago |
|
deit.py
|
Add DeiT-III 'medium' model defs and weights
|
2 years ago |
|
densenet.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
dla.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
dpn.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
edgenext.py
|
Add edgenext_base model def & weight link, update to improve ONNX export #1385
|
2 years ago |
|
efficientformer.py
|
Add EfficientFormer
|
2 years ago |
|
efficientnet.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
efficientnet_blocks.py
|
Merge remote-tracking branch 'origin/master' into norm_norm_norm
|
2 years ago |
|
efficientnet_builder.py
|
Merge remote-tracking branch 'origin/master' into norm_norm_norm
|
2 years ago |
|
factory.py
|
Revert ml-decoder changes to model factory and train script
|
2 years ago |
|
features.py
|
Significant model refactor and additions:
|
2 years ago |
|
fx_features.py
|
Update type hint for `register_notrace_module`
|
2 years ago |
|
gcvit.py
|
Add GCViT
|
2 years ago |
|
ghostnet.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
gluon_resnet.py
|
Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.
|
2 years ago |
|
gluon_xception.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
hardcorenas.py
|
Significant model refactor and additions:
|
2 years ago |
|
helpers.py
|
no longer a point using kwargs for pretrain_cfg resolve, just pass explicit arg
|
2 years ago |
|
hrnet.py
|
Ensure all model entrypoint fn default to `pretrained=False` (a few didn't)
|
2 years ago |
|
hub.py
|
👽 use hf_hub_download instead of cached_download
|
2 years ago |
|
inception_resnet_v2.py
|
Significant model refactor and additions:
|
2 years ago |
|
inception_v3.py
|
Improve resolve_pretrained_cfg behaviour when no cfg exists, warn instead of crash. Improve usability ex #1311
|
2 years ago |
|
inception_v4.py
|
Significant model refactor and additions:
|
2 years ago |
|
levit.py
|
Fix pit implementation to be clsoer to deit/levit re distillation head handling
|
2 years ago |
|
mlp_mixer.py
|
Change models with mean 0,0,0 std 1,1,1 from int to float for consistency as mentioned in #1355
|
2 years ago |
|
mobilenetv3.py
|
Change models with mean 0,0,0 std 1,1,1 from int to float for consistency as mentioned in #1355
|
2 years ago |
|
mobilevit.py
|
Change models with mean 0,0,0 std 1,1,1 from int to float for consistency as mentioned in #1355
|
2 years ago |
|
mvitv2.py
|
Add MViT (Multi-Scale) V2
|
2 years ago |
|
nasnet.py
|
Significant model refactor and additions:
|
2 years ago |
|
nest.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
nfnet.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
pit.py
|
Fix pit implementation to be clsoer to deit/levit re distillation head handling
|
2 years ago |
|
pnasnet.py
|
Significant model refactor and additions:
|
2 years ago |
|
poolformer.py
|
Remove some redundant requires_grad=True from nn.Parameter in third party code
|
2 years ago |
|
pvt_v2.py
|
Add PyramidVisionTransformerV2
|
2 years ago |
|
registry.py
|
Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.
|
2 years ago |
|
regnet.py
|
Fix group_matcher regex for regnet.py
|
2 years ago |
|
res2net.py
|
Significant model refactor and additions:
|
2 years ago |
|
resnest.py
|
Significant model refactor and additions:
|
2 years ago |
|
resnet.py
|
Ensure all model entrypoint fn default to `pretrained=False` (a few didn't)
|
2 years ago |
|
resnetv2.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
rexnet.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
selecsls.py
|
Significant model refactor and additions:
|
2 years ago |
|
senet.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
sequencer.py
|
Fix tests for rank-4 output where feature channels dim is -1 (3) and not 1
|
2 years ago |
|
sknet.py
|
Significant model refactor and additions:
|
2 years ago |
|
swin_transformer.py
|
Add initial swinv2 tiny / small weights
|
2 years ago |
|
swin_transformer_v2.py
|
Add updated vit_relpos weights, and impl w/ support for official swin-v2 differences for relpos. Add bias control support for MLP layers
|
2 years ago |
|
swin_transformer_v2_cr.py
|
Remove persistent buffers from Swin-V2. Change SwinV2Cr cos attn + tau/logit_scale to match official, add ckpt convert, init_value zeros resid LN weight by default
|
2 years ago |
|
tnt.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
tresnet.py
|
Change models with mean 0,0,0 std 1,1,1 from int to float for consistency as mentioned in #1355
|
2 years ago |
|
twins.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
vgg.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
visformer.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
vision_transformer.py
|
Take `no_emb_class` into account when calling `resize_pos_embed`
|
2 years ago |
|
vision_transformer_hybrid.py
|
Vision Transformer refactoring and Rel Pos impl
|
2 years ago |
|
vision_transformer_relpos.py
|
Update vit_relpos w/ some additional weights, some cleanup to match recent vit updates, more MLP log coord experiments.
|
2 years ago |
|
volo.py
|
Significant model refactor and additions:
|
2 years ago |
|
vovnet.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
xception.py
|
Significant model refactor and additions:
|
2 years ago |
|
xception_aligned.py
|
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
|
2 years ago |
|
xcit.py
|
Remove some redundant requires_grad=True from nn.Parameter in third party code
|
2 years ago |
Ross Wightman
6e559e9b5f