Skip to content

Commit db8ad25

Browse files
committed
MobileNetV3 appears correct based on paper update, cleaned up comments and compacted last block def
1 parent 20d66be commit db8ad25

File tree

2 files changed

+1
-5
lines changed

2 files changed

+1
-5
lines changed

models/genmobilenet.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -892,15 +892,11 @@ def _gen_mobilenet_v3(depth_multiplier, num_classes=1000, **kwargs):
892892
# stage 2, 56x56 in
893893
['ir_r3_k5_s2_e3_c40_se0.25_are'], # relu
894894
# stage 3, 28x28 in
895-
# FIXME are expansions here correct?
896895
['ir_r1_k3_s2_e6_c80', 'ir_r1_k3_s1_e2.5_c80', 'ir_r2_k3_s1_e2.3_c80'], # hard-swish
897896
# stage 4, 14x14in
898897
['ir_r2_k3_s1_e6_c112_se0.25'], # hard-swish
899898
# stage 5, 14x14in
900-
# FIXME paper has a mistaken block-stride pattern 1-2-1 that doesn't fit the usual 2-1-..., ignoring
901-
# The paper numbers result in an exp factor of 4.2 in the middle of this block, but keeping at 6
902-
# results in a param count closer to 5.4m
903-
['ir_r1_k5_s2_e6_c160_se0.25', 'ir_r1_k5_s1_e6_c160_se0.25', 'ir_r1_k5_s1_e6_c160_se0.25'], # hard-swish
899+
['ir_r3_k5_s2_e6_c160_se0.25'], # hard-swish
904900
# stage 6, 7x7 in
905901
['cn_r1_k1_s1_c960'], # hard-swish
906902
]

optim/adam_tf.py

Whitespace-only changes.

0 commit comments

Comments
 (0)