Ross Wightman
|
bcec14d3b5
|
Bring EfficientNet SE layer in line with others, pull se_ratio outside of blocks. Allows swapping w/ other attn layers.
|
2021-05-29 23:41:38 -07:00 |
|
Ross Wightman
|
9611458e19
|
Throw in some FBNetV3 code I had lying around, some refactoring of SE reduction channel calcs for all EffNet archs.
|
2021-05-28 20:47:24 -07:00 |
|
Ross Wightman
|
9c78de8c02
|
Fix #661, move hardswish out of default args for LeViT. Enable native torch support for hardswish, hardsigmoid, mish if present.
|
2021-05-26 15:28:42 -07:00 |
|
Ross Wightman
|
c4f482a08b
|
EfficientNetV2 official impl w/ weights ported from TF. Cleanup/refactor of related EfficientNet classes and models.
|
2021-05-14 15:50:00 -07:00 |
|
Ross Wightman
|
d5473c17f7
|
Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there.
|
2021-05-04 21:27:15 -07:00 |
|
Ross Wightman
|
2df77ee5cb
|
Fix torchscript compat and features_only behaviour in GhostNet PR. A few minor formatting changes. Reuse existing layers.
|
2021-04-15 10:20:26 -07:00 |
iamhankai
|
de445e7827
|
Add GhostNet
|
2021-04-13 23:19:51 +08:00 |