提交 91a4064e 编写于 作者: L liaogang

Follow comments

上级 724d6dd4
...@@ -1292,7 +1292,7 @@ def bilinear_interp_layer(input, ...@@ -1292,7 +1292,7 @@ def bilinear_interp_layer(input,
assert input.layer_type == LayerType.CONV_LAYER assert input.layer_type == LayerType.CONV_LAYER
assert isinstance(input.activation, LinearActivation) assert isinstance(input.activation, LinearActivation)
assert out_size_x > 0 and out_size_y > 0 assert out_size_x > 0 and out_size_y > 0
assert input.numfilters is not None assert input.num_filters is not None
num_channels = input.num_filters num_channels = input.num_filters
Layer(name=name, Layer(name=name,
inputs=Input(input.name, inputs=Input(input.name,
...@@ -1301,8 +1301,7 @@ def bilinear_interp_layer(input, ...@@ -1301,8 +1301,7 @@ def bilinear_interp_layer(input,
num_channels=num_channels)), num_channels=num_channels)),
type=LayerType.BILINEAR_INTERP_LAYER, type=LayerType.BILINEAR_INTERP_LAYER,
**ExtraLayerAttribute.to_kwargs(layer_attr)) **ExtraLayerAttribute.to_kwargs(layer_attr))
return LayerOutput(name, LayerType.BILINEAR_INTERP_LAYER, parents=[input], return LayerOutput(name, LayerType.BILINEAR_INTERP_LAYER, parents=[input])
num_filters=num_filters)
@wrap_name_default() @wrap_name_default()
@layer_support() @layer_support()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册