Ross Wightman 298fba09ac Back out some activation hacks trialing upcoming pytorch changes

2020-07-17 18:41:37 -07:00

..

__init__.py

Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces

2020-07-17 17:54:26 -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

Back out some activation hacks trialing upcoming pytorch changes

2020-07-17 18:41:37 -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

…

classifier.py

Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces

2020-07-17 17:54:26 -07:00

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

Back out some activation hacks trialing upcoming pytorch changes

2020-07-17 18:41:37 -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

Add initial impl of CrossStagePartial networks, yet to be trained, not quite the same as darknet cfgs.

2020-07-13 15:01:06 -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

Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces

2020-07-17 17:54:26 -07:00

inplace_abn.py

Add initial impl of CrossStagePartial networks, yet to be trained, not quite the same as darknet cfgs.

2020-07-13 15:01:06 -07:00

median_pool.py

…

mixed_conv2d.py

…

norm_act.py

Switch DPN to use BnAct layer, train a new DPN 68b model with RA to 79.21

2020-07-12 11:13:06 -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

Add initial impl of CrossStagePartial networks, yet to be trained, not quite the same as darknet cfgs.

2020-07-13 15:01:06 -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