Skip to content
This repository was archived by the owner on Apr 23, 2025. It is now read-only.

Commit 2ac021c

Browse files
authored
add mobilenet v3 (#409)
* add mobilenet v3 * fix filter shape mismatch * add some tests + lint * apply filter mult to input conv as well, unify styles * swap hard sigmoid back in * add hard sigmoid to initial block as well * add zeropad before initial block * add dropout * move stride up a layer * initial dropout rate tweak * refactor activation bool to enum type * refactor to use seblock layer * swift-format pass * fix se block conv type
1 parent 05ee525 commit 2ac021c

File tree

3 files changed

+514
-4
lines changed

3 files changed

+514
-4
lines changed

Models/ImageClassification/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ add_library(ImageClassificationModels
33
LeNet-5.swift
44
MobileNetV1.swift
55
MobileNetV2.swift
6+
MobileNetV3.swift
67
ResNet.swift
78
ResNetV2.swift
89
SqueezeNet.swift

0 commit comments

Comments
 (0)