pytorch-image-models

Commit Graph

Author	SHA1	Message	Date
Ross Wightman	5e6dbbaf30	Add CBAM for experimentation	4 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	4 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 (27b3680d49e0e7c30ae6dd25f0daeb8de5319b40)