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
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
4 years ago
..
__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.
4 years ago
activations.py
Monster commit, activation refactor, VoVNet, norm_act improvements, more
4 years ago
activations_jit.py
Monster commit, activation refactor, VoVNet, norm_act improvements, more
4 years ago
activations_me.py
Add custom grad tests, fix cut & paste error with hard_mish ME, add a few more pytorch act fns to factory
4 years ago
adaptive_avgmax_pool.py
…
anti_aliasing.py
…
blur_pool.py
…
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.
4 years ago
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.
4 years ago
conv2d_same.py
Monster commit, activation refactor, VoVNet, norm_act improvements, more
4 years ago
conv_bn_act.py
Monster commit, activation refactor, VoVNet, norm_act improvements, more
4 years ago
create_act.py
Add custom grad tests, fix cut & paste error with hard_mish ME, add a few more pytorch act fns to factory
4 years ago
create_attn.py
Monster commit, activation refactor, VoVNet, norm_act improvements, more
4 years ago
create_conv2d.py
Monster commit, activation refactor, VoVNet, norm_act improvements, more
4 years ago
create_norm_act.py
Monster commit, activation refactor, VoVNet, norm_act improvements, more
4 years ago
drop.py
Significant ResNet refactor:
4 years ago
eca.py
…
evo_norm.py
Monster commit, activation refactor, VoVNet, norm_act improvements, more
4 years ago
helpers.py
…
inplace_abn.py
Monster commit, activation refactor, VoVNet, norm_act improvements, more
4 years ago
median_pool.py
…
mixed_conv2d.py
…
norm_act.py
Fix an an untested change, remove a debug print
4 years ago
padding.py
…
pool2d_same.py
Fix a few issues that came up in tests
4 years ago
se.py
Fix two regressions
4 years ago
selective_kernel.py
Monster commit, activation refactor, VoVNet, norm_act improvements, more
4 years ago
separable_conv.py
Monster commit, activation refactor, VoVNet, norm_act improvements, more
4 years ago
space_to_depth.py
…
split_attn.py
…
split_batchnorm.py
…
test_time_pool.py
Monster commit, activation refactor, VoVNet, norm_act improvements, more
4 years ago
weight_init.py
…