pytorch-image-models

Commit Graph

Author	SHA1	Message	Date
Ross Wightman	5e6dbbaf30	Add CBAM for experimentation	6 years ago
Ross Wightman	f902bcd54c	Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases * select_conv2d -> create_conv2d * added create_attn to create attention module from string/bool/module * factor padding helpers into own file, use in both conv2d_same and avg_pool2d_same * add some more test eca resnet variants * minor tweaks, naming, comments, consistency	6 years ago

Author

SHA1

Message

Date

Ross Wightman

5e6dbbaf30

Add CBAM for experimentation

Ross Wightman

f902bcd54c

Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases

* select_conv2d -> create_conv2d
* added create_attn to create attention module from string/bool/module
* factor padding helpers into own file, use in both conv2d_same and avg_pool2d_same
* add some more test eca resnet variants
* minor tweaks, naming, comments, consistency

2 Commits (aecd7b795d8d6d6fa19ae7db95bcc73c2900dd49)