pytorch-image-models

History

Ross Wightman 803254bb40 Fix spacing misalignment for fast norm path in LayerNorm modules		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 norm/norm_act header comments	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	Fix spacing misalignment for fast norm path in LayerNorm modules	2 years ago
norm_act.py	Add norm/norm_act header comments	2 years ago
padding.py	…
patch_embed.py	Adding support for fine-tune CLIP LAION-2B image tower weights for B/32, L/14, H/14 and g/14. Still WIP	2 years ago
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 CL SE module	2 years ago
std_conv.py	…
test_time_pool.py	…
trace_utils.py	…
weight_init.py	Clean up no_grad for trunc normal weight inits	2 years ago