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/layers
Ross Wightman 9d03c6f526
Merge remote-tracking branch 'origin/main' into levit_efficientformer_redux
1 year ago
..
__init__.py Factor NormMlpClassifierHead from MaxxViT and use across MaxxViT / ConvNeXt / DaViT, refactor some type hints & comments 1 year ago
activations.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
activations_jit.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
activations_me.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
adaptive_avgmax_pool.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
attention_pool2d.py Add FlexiViT models and weights, refactoring, push more weights 1 year ago
blur_pool.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
bottleneck_attn.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
cbam.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
classifier.py Factor NormMlpClassifierHead from MaxxViT and use across MaxxViT / ConvNeXt / DaViT, refactor some type hints & comments 1 year ago
cond_conv2d.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
config.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
conv2d_same.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
conv_bn_act.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
create_act.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
create_attn.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
create_conv2d.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
create_norm.py Add EfficientFormer-V2, refactor EfficientFormer and Levit for more uniformity across the 3 related arch. Add features_out support to levit conv models and efficientformer_v2. All weights on hub. 1 year ago
create_norm_act.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
drop.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
eca.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
evo_norm.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
fast_norm.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
filter_response_norm.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
gather_excite.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
global_context.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
grn.py Add ConvNeXt-V2 support (model additions and weights) (#1614) 1 year ago
halo_attn.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
helpers.py Add FlexiViT models and weights, refactoring, push more weights 1 year ago
inplace_abn.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
lambda_layer.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
linear.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
median_pool.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
mixed_conv2d.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
ml_decoder.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
mlp.py Add ConvNeXt-V2 support (model additions and weights) (#1614) 1 year ago
non_local_attn.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
norm.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
norm_act.py Update batchnorm freezing to handle NormAct variants, Add GroupNorm1Act, update BatchNormAct2d tracing change from PyTorch 1 year ago
padding.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
patch_embed.py Better vmap compat across recent torch versions 1 year ago
pool2d_same.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
pos_embed.py Add FlexiViT models and weights, refactoring, push more weights 1 year ago
pos_embed_rel.py Add FlexiViT models and weights, refactoring, push more weights 1 year ago
pos_embed_sincos.py Add FlexiViT models and weights, refactoring, push more weights 1 year ago
selective_kernel.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
separable_conv.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
space_to_depth.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
split_attn.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
split_batchnorm.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
squeeze_excite.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
std_conv.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
test_time_pool.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
trace_utils.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
weight_init.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago