Ross Wightman
|
110a7c4982
|
AdaptiveAvgPool2d -> mean((2,3)) for all SE/attn layers to avoid NaN with AMP + channels_last layout. See https://github.com/pytorch/pytorch/issues/43992
|
2020-09-01 16:05:32 -07:00 |
|
Ross Wightman
|
6c17d57a2c
|
Fix some attributions, add copyrights to some file docstrings
|
2020-07-27 13:44:56 -07:00 |
|
Ross Wightman
|
569419b38d
|
Tweak some comments, add SKNet models with weights to sotabench, remove an unused branch
|
2020-02-15 21:18:25 -08:00 |
|
Ross Wightman
|
5e6dbbaf30
|
Add CBAM for experimentation
|
2020-02-10 16:23:09 -08:00 |