未验证 提交 33e9e7e2 编写于 作者: B Bai Yifan 提交者: GitHub

fix prefix name (#3459)

上级 ac5b7971
......@@ -77,6 +77,7 @@ class ResNet():
pool = fluid.layers.pool2d(
input=conv, pool_size=7, pool_type='avg', global_pooling=True)
stdv = 1.0 / math.sqrt(pool.shape[1] * 1.0)
fc_name = fc_name if fc_name is None else prefix_name + fc_name
out = fluid.layers.fc(input=pool,
size=class_dim,
act='softmax',
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册