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 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	Merge pull request #282 from tigert1998/patch-1	4 years ago
adaptive_avgmax_pool.py	…
attention_pool2d.py	…
blur_pool.py	Make a few more layers symbolically traceable (remove from FX leaf modules)	3 years ago
bottleneck_attn.py	wip - pre-rebase	3 years ago
cbam.py	…
classifier.py	…
cond_conv2d.py	…
config.py	Add set_layer_config contextmgr to adjust all layer configs at once, use in create_module with new args. Remove a few old warning causing constant annotations for jit.	4 years ago
conv2d_same.py	…
conv_bn_act.py	Add more experimental darknet and 'cs2' darknet variants (different cross stage setup, closer to newer YOLO backbones) for train trials.	2 years ago
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	Improve repr for DropPath module	2 years ago
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	Significant norm update	3 years ago
lambda_layer.py	wip - pre-rebase	3 years ago
linear.py	…
median_pool.py	Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers.	4 years ago
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	Fix some attributions, add copyrights to some file docstrings	4 years ago
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	Fix some attributions, add copyrights to some file docstrings	4 years ago
squeeze_excite.py	Add CL SE module	2 years ago
std_conv.py	…
test_time_pool.py	…
trace_utils.py	Better fix for #954 that doesn't break torchscript, pull torch._assert into timm namespace when it exists	3 years ago
weight_init.py	Clean up no_grad for trunc normal weight inits	2 years ago