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/models/layers
Ross Wightman ffaf97f813
MaxxVit! A very configurable MaxVit and CoAtNet impl with lots of goodies..
2 years ago
..
__init__.py MaxxVit! A very configurable MaxVit and CoAtNet impl with lots of goodies.. 2 years ago
activations.py
activations_jit.py
activations_me.py
adaptive_avgmax_pool.py
attention_pool2d.py
blur_pool.py
bottleneck_attn.py
cbam.py
classifier.py
cond_conv2d.py
config.py
conv2d_same.py
conv_bn_act.py
create_act.py 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
create_attn.py
create_conv2d.py
create_norm.py 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
create_norm_act.py More model and test fixes 2 years ago
drop.py
eca.py
evo_norm.py
fast_norm.py 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
filter_response_norm.py
gather_excite.py
global_context.py
halo_attn.py
helpers.py MaxxVit! A very configurable MaxVit and CoAtNet impl with lots of goodies.. 2 years ago
inplace_abn.py
lambda_layer.py
linear.py
median_pool.py
mixed_conv2d.py
ml_decoder.py
mlp.py
non_local_attn.py
norm.py 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
norm_act.py More model and test fixes 2 years ago
padding.py
patch_embed.py
pool2d_same.py
pos_embed.py
selective_kernel.py
separable_conv.py
space_to_depth.py
split_attn.py
split_batchnorm.py
squeeze_excite.py 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
std_conv.py
test_time_pool.py
trace_utils.py
weight_init.py