Skip to content

Commit 47d40bf

Browse files
authored
fix ShuffleNetV2 config (PaddlePaddle#6138)
* fix ShuffleNetV2 x2_0 stage_out_channels * fix format, test=document_fix
1 parent 2f06ad8 commit 47d40bf

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

ppdet/modeling/backbones/shufflenet_v2.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -188,11 +188,10 @@ def __init__(self, scale=1.0, act="relu", feature_maps=[5, 13, 17]):
188188
elif scale == 1.5:
189189
stage_out_channels = [-1, 24, 176, 352, 704, 1024]
190190
elif scale == 2.0:
191-
stage_out_channels = [-1, 24, 224, 488, 976, 2048]
191+
stage_out_channels = [-1, 24, 244, 488, 976, 2048]
192192
else:
193193
raise NotImplementedError("This scale size:[" + str(scale) +
194194
"] is not implemented!")
195-
196195
self._out_channels = []
197196
self._feature_idx = 0
198197
# 1. conv1

0 commit comments

Comments
 (0)