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.

Ross Wightman f670d98cb8 Make a few more layers symbolically traceable (remove from FX leaf modules) ... * remove dtype kwarg from .to() calls in EvoNorm as it messed up script + trace combo * BatchNormAct2d always uses custom forward (cut & paste from original) instead of super().forward. Fixes #1176 * BlurPool groups==channels, no need to use input.dim[1]		3 years ago
..
layers	Make a few more layers symbolically traceable (remove from FX leaf modules)	3 years ago
pruned	added efficientnet pruned weights	5 years ago
__init__.py	Merge remote-tracking branch 'origin/master' into norm_norm_norm	3 years ago
beit.py	Significant model refactor and additions:	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 ConvNeXt tiny and small pretrain in22k	3 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	Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet	3 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
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	Fix pruned adapt for EfficientNet models that are now using BatchNormAct layers	3 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	Significant model refactor and additions:	3 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	Significant model refactor and additions:	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	Update test crop for new RegNet-V weights to match Y	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	Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet	3 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
sknet.py	Significant model refactor and additions:	3 years ago
swin_transformer.py	Add attrib / comments about Swin-S3 (AutoFormerV2) weights	3 years ago
swin_transformer_v2_cr.py	version 0.6.1 for master	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	Add layer scale and parallel blocks to vision_transformer	3 years ago
vision_transformer_hybrid.py	Merge remote-tracking branch 'origin/master' into norm_norm_norm	3 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