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 4d5c395160 MaxVit, ViT, ConvNeXt, and EfficientNet-v2 updates ... * Add support for TF weights and modelling specifics to MaxVit (testing ported weights) * More fine-tuned CLIP ViT configs * ConvNeXt and MaxVit updated to new pretrained cfgs use * EfficientNetV2, MaxVit and ConvNeXt high res models use squash crop/resize		3 years ago
..
__init__.py	MaxxVit! A very configurable MaxVit and CoAtNet impl with lots of goodies..	3 years ago
activations.py	MaxVit, ViT, ConvNeXt, and EfficientNet-v2 updates	3 years ago
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	MaxVit, ViT, ConvNeXt, and EfficientNet-v2 updates	3 years ago
create_attn.py	Add MobileVitV2 support. Fix #1332. Move GroupNorm1 to common layers (used in poolformer + mobilevitv2). Keep ol custom ConvNeXt LayerNorm2d impl as LayerNormExp2d for reference.	3 years ago
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)	3 years ago
create_norm_act.py	More model and test fixes	3 years ago
drop.py	…
eca.py	…
evo_norm.py	…
fast_norm.py	Add norm/norm_act header comments	3 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..	3 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	3 years ago
norm_act.py	Add norm/norm_act header comments	3 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	3 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	3 years ago
std_conv.py	…
test_time_pool.py	Add --use-train-size flag to force use of train input_size (over test input size) for validation. Default test-time pooling to use train input size (fixes issues).	3 years ago
trace_utils.py	…
weight_init.py	Clean up no_grad for trunc normal weight inits	3 years ago