Skip to content

Commit 7a6d615

Browse files
committed
Add EfficientNet impl, change existing depth_multipler -> channel_multiplier as definitions have been confused
1 parent 6bff9c7 commit 7a6d615

File tree

2 files changed

+266
-125
lines changed

2 files changed

+266
-125
lines changed

README.md

+1
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ I've included a few of my favourite models, but this is not an exhaustive collec
3030
* DPN (from [me](https://github.com/rwightman/pytorch-dpn-pretrained), weights hosted by Cadene)
3131
* DPN-68, DPN-68b, DPN-92, DPN-98, DPN-131, DPN-107
3232
* Generic MobileNet (from my standalone [GenMobileNet](https://github.com/rwightman/genmobilenet-pytorch)) - A generic model that implements many of the mobile optimized architecture search derived models that utilize similar DepthwiseSeparable and InvertedResidual blocks
33+
* EfficientNet (B0-B4) (https://arxiv.org/abs/1905.11946) -- work in progress, validating
3334
* MNASNet B1, A1 (Squeeze-Excite), and Small (https://arxiv.org/abs/1807.11626)
3435
* MobileNet-V1 (https://arxiv.org/abs/1704.04861)
3536
* MobileNet-V2 (https://arxiv.org/abs/1801.04381)

0 commit comments

Comments
 (0)