Ross Wightman 8bf63b6c6c Able to use other attn layer in EfficientNet now. Create test ECA + GC B0 configs. Make ECA more configurable.		2021-05-30 12:47:02 -07:00
..
__init__.py	Add Gather-Excite and Global Context attn modules. Refactor existing SE-like attn for consistency and refactor byob/byoanet for less redundancy.	2021-05-27 18:03:29 -07:00
activations.py	Fix inplace arg compat for GELU and PreLU via activation factory	2020-11-30 13:27:40 -08:00
activations_jit.py	…
activations_me.py	Merge pull request #282 from tigert1998/patch-1	2021-02-04 12:18:40 -08:00
adaptive_avgmax_pool.py	…
blur_pool.py	Remove filter hack from BlurPool w/ non-persistent buffer. Use BlurPool2d instead of AntiAliasing.. for TResNet. Breaks PyTorch < 1.6.	2021-05-04 16:56:28 -07:00
bottleneck_attn.py	Improved (hopefully) init for SA/SA-like layers used in ByoaNets	2021-05-04 21:40:39 -07:00
cbam.py	Add Gather-Excite and Global Context attn modules. Refactor existing SE-like attn for consistency and refactor byob/byoanet for less redundancy.	2021-05-27 18:03:29 -07:00
classifier.py	ResNetV2 pre-act and non-preact model, w/ BiT pretrained weights and support for ViT R50 model. Tweaks for in21k num_classes passing. More to do... tests failing.	2020-12-28 16:59:15 -08:00
cond_conv2d.py	Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers.	2020-10-13 13:33:44 -07:00
config.py	…
conv2d_same.py	…
conv_bn_act.py	Fix regression in models with 1001 class pretrained weights. Improve batchnorm arg and BatchNormAct layer handling in several models.	2021-02-09 16:22:52 -08:00
create_act.py	Fix #661, move hardswish out of default args for LeViT. Enable native torch support for hardswish, hardsigmoid, mish if present.	2021-05-26 15:28:42 -07:00
create_attn.py	Add Gather-Excite and Global Context attn modules. Refactor existing SE-like attn for consistency and refactor byob/byoanet for less redundancy.	2021-05-27 18:03:29 -07:00
create_conv2d.py	Use in_channels for depthwise groups, allows using `out_channels=N * in_channels` (does not impact existing models). Fix #354.	2021-02-09 16:22:52 -08:00
create_norm_act.py	Fix regression in models with 1001 class pretrained weights. Improve batchnorm arg and BatchNormAct layer handling in several models.	2021-02-09 16:22:52 -08:00
create_self_attn.py	Add initial RedNet model / Involution layer impl for testing	2021-05-14 17:16:34 -07:00
drop.py	Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers.	2020-10-13 13:33:44 -07:00
eca.py	Able to use other attn layer in EfficientNet now. Create test ECA + GC B0 configs. Make ECA more configurable.	2021-05-30 12:47:02 -07:00
evo_norm.py	…
gather_excite.py	Add Gather-Excite and Global Context attn modules. Refactor existing SE-like attn for consistency and refactor byob/byoanet for less redundancy.	2021-05-27 18:03:29 -07:00
global_context.py	Add Gather-Excite and Global Context attn modules. Refactor existing SE-like attn for consistency and refactor byob/byoanet for less redundancy.	2021-05-27 18:03:29 -07:00
halo_attn.py	Improved (hopefully) init for SA/SA-like layers used in ByoaNets	2021-05-04 21:40:39 -07:00
helpers.py	Throw in some FBNetV3 code I had lying around, some refactoring of SE reduction channel calcs for all EffNet archs.	2021-05-28 20:47:24 -07:00
inplace_abn.py	Update README, fix iabn pip version print.	2021-03-07 16:17:06 -08:00
involution.py	Add Gather-Excite and Global Context attn modules. Refactor existing SE-like attn for consistency and refactor byob/byoanet for less redundancy.	2021-05-27 18:03:29 -07:00
lambda_layer.py	Improved (hopefully) init for SA/SA-like layers used in ByoaNets	2021-05-04 21:40:39 -07:00
linear.py	A few more changes for 0.3.2 maint release. Linear layer change for mobilenetv3 and inception_v3, support no bias for linear wrapper.	2020-11-30 16:19:52 -08:00
median_pool.py	Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers.	2020-10-13 13:33:44 -07:00
mixed_conv2d.py	Use in_channels for depthwise groups, allows using `out_channels=N * in_channels` (does not impact existing models). Fix #354.	2021-02-09 16:22:52 -08:00
mlp.py	Add Gather-Excite and Global Context attn modules. Refactor existing SE-like attn for consistency and refactor byob/byoanet for less redundancy.	2021-05-27 18:03:29 -07:00
norm.py	Add Gather-Excite and Global Context attn modules. Refactor existing SE-like attn for consistency and refactor byob/byoanet for less redundancy.	2021-05-27 18:03:29 -07:00
norm_act.py	Fix regression in models with 1001 class pretrained weights. Improve batchnorm arg and BatchNormAct layer handling in several models.	2021-02-09 16:22:52 -08:00
padding.py	…
patch_embed.py	Add levit, levit_c, and visformer model defs. Largely untested and not finished cleanup.	2021-05-14 17:16:34 -07:00
pool2d_same.py	Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers.	2020-10-13 13:33:44 -07:00
selective_kernel.py	…
separable_conv.py	Fix regression in models with 1001 class pretrained weights. Improve batchnorm arg and BatchNormAct layer handling in several models.	2021-02-09 16:22:52 -08:00
space_to_depth.py	…
split_attn.py	…
split_batchnorm.py	…
squeeze_excite.py	Add Gather-Excite and Global Context attn modules. Refactor existing SE-like attn for consistency and refactor byob/byoanet for less redundancy.	2021-05-27 18:03:29 -07:00
std_conv.py	Add ECA-NFNet-L0 weights and update model name. Update README and bump version to 0.4.6	2021-03-17 13:55:32 -07:00
swin_attn.py	Improved (hopefully) init for SA/SA-like layers used in ByoaNets	2021-05-04 21:40:39 -07:00
test_time_pool.py	Add new weights for ecaresnet26t/50t/269d models. Remove distinction between 't' and 'tn' (tiered models), tn is now t. Add test time img size spec to default cfg.	2021-02-06 16:30:02 -08:00
weight_init.py	Cleanup experimental vit weight init a bit	2021-03-20 09:44:24 -07:00