Skip to content

S
seminar-breakout
Commit 17955e89 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix #862

parent 92b4d3bf
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/ImageNetModels/shufflenet.py
View file @ 17955e89
...@@ -86,7 +86,7 @@ def shufflenet_unit_v2(l, out_channel, stride): ...@@ -86,7 +86,7 @@ def shufflenet_unit_v2(l, out_channel, stride):
shortcut, l = tf.split(l, 2, axis=1) shortcut, l = tf.split(l, 2, axis=1)
else: else:
shortcut, l = l, l shortcut, l = l, l
shortcut_channel = shortcut.shape[1] shortcut_channel = int(shortcut.shape[1])
l = Conv2D('conv1', l, out_channel // 2, 1, activation=BNReLU) l = Conv2D('conv1', l, out_channel // 2, 1, activation=BNReLU)
l = DepthConv('dconv', l, out_channel // 2, 3, stride=stride) l = DepthConv('dconv', l, out_channel // 2, 3, stride=stride)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment