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 e520553e3e
Update batchnorm freezing to handle NormAct variants, Add GroupNorm1Act, update BatchNormAct2d tracing change from PyTorch
2 years ago
..
__init__.py Update batchnorm freezing to handle NormAct variants, Add GroupNorm1Act, update BatchNormAct2d tracing change from PyTorch 2 years ago
activations.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
activations_jit.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
activations_me.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
adaptive_avgmax_pool.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
attention_pool2d.py Add FlexiViT models and weights, refactoring, push more weights 2 years ago
blur_pool.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
bottleneck_attn.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
cbam.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
classifier.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
cond_conv2d.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
config.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
conv2d_same.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
conv_bn_act.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
create_act.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
create_attn.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
create_conv2d.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
create_norm.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
create_norm_act.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
drop.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
eca.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
evo_norm.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
fast_norm.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
filter_response_norm.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
gather_excite.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
global_context.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
grn.py Add ConvNeXt-V2 support (model additions and weights) (#1614) 2 years ago
halo_attn.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
helpers.py Add FlexiViT models and weights, refactoring, push more weights 2 years ago
inplace_abn.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
lambda_layer.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
linear.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
median_pool.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
mixed_conv2d.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
ml_decoder.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
mlp.py Add ConvNeXt-V2 support (model additions and weights) (#1614) 2 years ago
non_local_attn.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
norm.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
norm_act.py Update batchnorm freezing to handle NormAct variants, Add GroupNorm1Act, update BatchNormAct2d tracing change from PyTorch 2 years ago
padding.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
patch_embed.py Better vmap compat across recent torch versions 2 years ago
pool2d_same.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
pos_embed.py Add FlexiViT models and weights, refactoring, push more weights 2 years ago
pos_embed_rel.py Add FlexiViT models and weights, refactoring, push more weights 2 years ago
pos_embed_sincos.py Add FlexiViT models and weights, refactoring, push more weights 2 years ago
selective_kernel.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
separable_conv.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
space_to_depth.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
split_attn.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
split_batchnorm.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
squeeze_excite.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
std_conv.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
test_time_pool.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
trace_utils.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
weight_init.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago