提交 a784c49b authored 作者: affanv14's avatar affanv14

fix small error when shapes are not given

上级 5b7740c7
...@@ -646,12 +646,15 @@ def separable_conv2d(input, ...@@ -646,12 +646,15 @@ def separable_conv2d(input,
filter_dilation=filter_dilation, filter_dilation=filter_dilation,
num_groups=num_channels) num_groups=num_channels)
depthwise_op_shape = conv_op.infer_shape(None, [input_shape, depthwise_filter_shape]) if input_shape is None or depthwise_filter_shape is None:
depthwise_op_shape = None
else:
depthwise_op_shape = conv_op.infer_shape(None, [input_shape, depthwise_filter_shape])[0]
depthwise_op = conv_op(input, depthwise_filters) depthwise_op = conv_op(input, depthwise_filters)
pointwise_op = conv2d(input=depthwise_op, pointwise_op = conv2d(input=depthwise_op,
filters=pointwise_filters, filters=pointwise_filters,
input_shape=depthwise_op_shape[0], input_shape=depthwise_op_shape,
filter_shape=pointwise_filter_shape, filter_shape=pointwise_filter_shape,
border_mode='valid', border_mode='valid',
subsample=(1, 1), subsample=(1, 1),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论