..
layers
Add eps arg to LayerNorm2d, add 'tf' (tensorflow) variant of trunc_normal_ that applies scale/shift after sampling (instead of needing to move a/b)
2 years ago
pruned
added efficientnet pruned weights
5 years ago
__init__.py
Add initial EdgeNeXt import. Significant cleanup / reorg (like ConvNeXt). Fix #1320
2 years ago
beit.py
migrate azure blob for beit checkpoints
3 years ago
byoanet.py
Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.
3 years ago
byobnet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
cait.py
Significant model refactor and additions:
3 years ago
coat.py
Significant model refactor and additions:
3 years ago
convit.py
Significant model refactor and additions:
3 years ago
convmixer.py
Significant model refactor and additions:
3 years ago
convnext.py
Add initial EdgeNeXt import. Significant cleanup / reorg (like ConvNeXt). Fix #1320
2 years ago
crossvit.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
cspnet.py
Add more experimental darknet and 'cs2' darknet variants (different cross stage setup, closer to newer YOLO backbones) for train trials.
2 years ago
deit.py
Fix pit implementation to be clsoer to deit/levit re distillation head handling
3 years ago
densenet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
dla.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
dpn.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
edgenext.py
Add initial EdgeNeXt import. Significant cleanup / reorg (like ConvNeXt). Fix #1320
2 years ago
efficientnet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
efficientnet_blocks.py
Merge remote-tracking branch 'origin/master' into norm_norm_norm
3 years ago
efficientnet_builder.py
Merge remote-tracking branch 'origin/master' into norm_norm_norm
3 years ago
factory.py
Revert ml-decoder changes to model factory and train script
3 years ago
features.py
Significant model refactor and additions:
3 years ago
fx_features.py
Make a few more layers symbolically traceable (remove from FX leaf modules)
3 years ago
ghostnet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 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.
3 years ago
gluon_xception.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
hardcorenas.py
Significant model refactor and additions:
3 years ago
helpers.py
no longer a point using kwargs for pretrain_cfg resolve, just pass explicit arg
2 years ago
hrnet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
hub.py
Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.
3 years ago
inception_resnet_v2.py
Significant model refactor and additions:
3 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:
3 years ago
levit.py
Fix pit implementation to be clsoer to deit/levit re distillation head handling
3 years ago
mlp_mixer.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
mobilenetv3.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
mobilevit.py
Significant model refactor and additions:
3 years ago
nasnet.py
Significant model refactor and additions:
3 years ago
nest.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
nfnet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
pit.py
Fix pit implementation to be clsoer to deit/levit re distillation head handling
3 years ago
pnasnet.py
Significant model refactor and additions:
3 years ago
poolformer.py
Fix poolformer group_matcher to merge proj downsample with previous block, support coarse
3 years ago
registry.py
Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.
3 years ago
regnet.py
Fix group_matcher regex for regnet.py
3 years ago
res2net.py
Significant model refactor and additions:
3 years ago
resnest.py
Significant model refactor and additions:
3 years ago
resnet.py
Add resnet10t (basic block) and resnet14t (bottleneck) with 1,1,1,1 repeats
2 years ago
resnetv2.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
rexnet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
selecsls.py
Significant model refactor and additions:
3 years ago
senet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
sequencer.py
Fix tests for rank-4 output where feature channels dim is -1 (3) and not 1
3 years ago
sknet.py
Significant model refactor and additions:
3 years ago
swin_transformer.py
Add initial swinv2 tiny / small weights
3 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
3 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
3 years ago
tnt.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
tresnet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
twins.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
vgg.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
visformer.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
vision_transformer.py
Improve resolve_pretrained_cfg behaviour when no cfg exists, warn instead of crash. Improve usability ex #1311
2 years ago
vision_transformer_hybrid.py
Vision Transformer refactoring and Rel Pos impl
3 years ago
vision_transformer_relpos.py
Improve resolve_pretrained_cfg behaviour when no cfg exists, warn instead of crash. Improve usability ex #1311
2 years ago
volo.py
Significant model refactor and additions:
3 years ago
vovnet.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
xception.py
Significant model refactor and additions:
3 years ago
xception_aligned.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
xcit.py
Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet
3 years ago
Ross Wightman
6064d16a2d