未验证 提交 081b5dc7 编写于 作者: R ruri 提交者: GitHub

remove useless comments (#2189)

上级 d5ccd221
...@@ -82,7 +82,6 @@ class ShuffleNetV2(): ...@@ -82,7 +82,6 @@ class ShuffleNetV2():
use_cudnn=True, use_cudnn=True,
if_act=True, if_act=True,
name=None): name=None):
# print(num_groups)
conv = fluid.layers.conv2d( conv = fluid.layers.conv2d(
input=input, input=input,
num_filters=num_filters, num_filters=num_filters,
...@@ -95,8 +94,6 @@ class ShuffleNetV2(): ...@@ -95,8 +94,6 @@ class ShuffleNetV2():
param_attr=ParamAttr(initializer=MSRA(),name=name+'_weights'), param_attr=ParamAttr(initializer=MSRA(),name=name+'_weights'),
bias_attr=False) bias_attr=False)
out = int((input.shape[2] - 1)/float(stride) + 1) out = int((input.shape[2] - 1)/float(stride) + 1)
# print(input.shape[1],(out, out), num_filters, (filter_size, filter_size), stride,
# (filter_size - 1) / 2, num_groups, name)
bn_name = name + '_bn' bn_name = name + '_bn'
if if_act: if if_act:
return fluid.layers.batch_norm(input=conv, act='swish', return fluid.layers.batch_norm(input=conv, act='swish',
...@@ -137,8 +134,6 @@ class ShuffleNetV2(): ...@@ -137,8 +134,6 @@ class ShuffleNetV2():
if benchmodel == 1: if benchmodel == 1:
x1, x2 = fluid.layers.split( x1, x2 = fluid.layers.split(
input, num_or_sections=[input.shape[1]//2, input.shape[1]//2], dim=1) input, num_or_sections=[input.shape[1]//2, input.shape[1]//2], dim=1)
# x1 = input[:, :(input.shape[1]//2), :, :]
# x2 = input[:, (input.shape[1]//2):, :, :]
conv_pw = self.conv_bn_layer( conv_pw = self.conv_bn_layer(
input=x2, input=x2,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册