Ross Wightman c4f482a08b EfficientNetV2 official impl w/ weights ported from TF. Cleanup/refactor of related EfficientNet classes and models.		2021-05-14 15:50:00 -07:00
..
__init__.py	Move Mlp and PatchEmbed modules into layers. Being used in lots of models now...	2021-05-06 14:03:23 -07:00
activations.py	…
activations_jit.py	…
activations_me.py	…
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	…
classifier.py	…
cond_conv2d.py	…
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	…
create_attn.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_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	Fixup byoanet configs to pass unit tests. Add swin_attn and swinnet26t model for testing.	2021-04-29 21:08:37 -07:00
drop.py	…
eca.py	…
evo_norm.py	…
halo_attn.py	Improved (hopefully) init for SA/SA-like layers used in ByoaNets	2021-05-04 21:40:39 -07:00
helpers.py	EfficientNetV2 official impl w/ weights ported from TF. Cleanup/refactor of related EfficientNet classes and models.	2021-05-14 15:50:00 -07:00
inplace_abn.py	Update README, fix iabn pip version print.	2021-03-07 16:17:06 -08: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	…
median_pool.py	…
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	Move Mlp and PatchEmbed modules into layers. Being used in lots of models now...	2021-05-06 14:03:23 -07:00
norm.py	Missed norm.py	2021-04-12 09:57:56 -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	Rethink name of patch embed grid info	2021-05-06 14:08:20 -07:00
pool2d_same.py	…
se.py	…
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	…
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	…
weight_init.py	Cleanup experimental vit weight init a bit	2021-03-20 09:44:24 -07:00