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 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	…
create_attn.py	…
create_conv2d.py	…
create_norm.py	…
create_norm_act.py	…
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	…
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	…
std_conv.py	…
test_time_pool.py	…
trace_utils.py	…
weight_init.py	…