Ross Wightman f122f0274b Significant ResNet refactor: ...

* stage creation + make_layer moved to separate fn with more sensible dilation/output_stride calc
* drop path rate decay easy to impl with refactored block creation loops
* fix dilation + blur pool combo

2020-07-05 00:48:12 -07:00

..

__init__.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.

2020-06-02 21:06:10 -07:00

activations_jit.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

activations_me.py

Add custom grad tests, fix cut & paste error with hard_mish ME, add a few more pytorch act fns to factory

2020-06-11 14:49:23 -07:00

activations.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

adaptive_avgmax_pool.py

…

anti_aliasing.py

Final blurpool2d cleanup and add resnetblur50 weights, match tresnet Downsample arg order to BlurPool2d for interop

2020-05-01 17:00:21 -07:00

blur_pool.py

Final blurpool2d cleanup and add resnetblur50 weights, match tresnet Downsample arg order to BlurPool2d for interop

2020-05-01 17:00:21 -07:00

cbam.py

…

cond_conv2d.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.

2020-06-02 21:06:10 -07:00

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.

2020-06-02 21:06:10 -07:00

conv2d_same.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

conv_bn_act.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

create_act.py

Add custom grad tests, fix cut & paste error with hard_mish ME, add a few more pytorch act fns to factory

2020-06-11 14:49:23 -07:00

create_attn.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

create_conv2d.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

create_norm_act.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

drop.py

Significant ResNet refactor:

2020-07-05 00:48:12 -07:00

eca.py

…

evo_norm.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

helpers.py

…

inplace_abn.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

median_pool.py

…

mixed_conv2d.py

…

norm_act.py

Fix an an untested change, remove a debug print

2020-06-01 17:26:42 -07:00

padding.py

…

pool2d_same.py

Fix a few issues that came up in tests

2020-06-29 21:13:21 -07:00

se.py

Fix two regressions

2020-06-05 11:04:51 -07:00

selective_kernel.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

separable_conv.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

space_to_depth.py

…

split_attn.py

Improve dropblock impl, add fast variant, and better AMP speed, inplace, batchwise... few ResNeSt cleanups

2020-05-13 15:17:08 -07:00

split_batchnorm.py

…

test_time_pool.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

weight_init.py

Remove annoying InceptionV3 dependency on scipy and insanely slow trunc_norm init. Bring InceptionV3 code into this codebase and use upcoming torch trunch_norm_ init.

2020-05-12 21:59:34 -07:00