提交 ad957c26 编写于 作者: Y Yibing Liu

Correct two parameters

上级 4a0109ff
......@@ -125,7 +125,7 @@ def stem(input):
filter_size=3,
stride=1,
padding=1)
return fluid.layers.concat(input=[l_conv1, r_conv3], axis=3)
return fluid.layers.concat(input=[l_conv1, r_conv3], axis=1)
def block2(input):
conv0 = conv_bn_layer(
......
......@@ -222,7 +222,7 @@ def train_parallel_exe(args,
use_nccl=True,
lr_strategy=None,
layers=50):
class_dim = 101
class_dim = 1000
image_shape = [3, 224, 224]
image = fluid.layers.data(name='image', shape=image_shape, dtype='float32')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册