提交 0152d97e 编写于 作者: W wanghaoshuang

fix pixel softmax python wrapper bug

上级 3aa67981
......@@ -3176,7 +3176,7 @@ class RecurrentLayerGroup(LayerBase):
@config_layer('pixel_softmax')
class PixelSoftmaxLayer(LayerBase):
def __init__(self, input, name, **xargs):
def __init__(self, name, inputs, **xargs):
super(PixelSoftmaxLayer, self).__init__(
name, 'pixel_softmax', 0, inputs=inputs, **xargs)
......
......@@ -126,6 +126,7 @@ __all__ = [
'row_conv_layer',
'dropout_layer',
'prelu_layer',
'pixel_softmax_layer',
]
......@@ -5905,8 +5906,8 @@ def pixel_softmax_layer(input, name=None, layer_attr=None):
else:
assert isinstance(input, collections.Sequence)
l = Layer(
inputs=[x.name for x in input],
name=name,
inputs=[x.name for x in input],
type=LayerType.PIXEL_SOFTMAX_LAYER,
**ExtraLayerAttribute.to_kwargs(layer_attr))
return LayerOutput(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册