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

fix pixel softmax python wrapper bug

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