提交 16023300 编写于 作者: L liuqi

Fix validation tool bug.

上级 8b18778a
...@@ -477,11 +477,22 @@ class CaffeConverter(object): ...@@ -477,11 +477,22 @@ class CaffeConverter(object):
else: else:
op_def.input.extend([bias_tensor_name]) op_def.input.extend([bias_tensor_name])
self.resolved_ops.add(op.name)
output_shape = Shapes.fully_connected_shape(input_shape, weight_data.shape) output_shape = Shapes.fully_connected_shape(input_shape, weight_data.shape)
op.output_shape = output_shape op.output_shape = output_shape
final_op = op
self.resolved_ops.add(op.name) if len(self.ops_map[final_op.name].children) == 1 \
op_def.output.extend([op.name + ':0']) and self.ops_map[final_op.name].children[0].type in activation_name_map:
activation_op = self.ops_map[final_op.name].children[0]
fused_act_arg = op_def.arg.add()
fused_act_arg.name = 'activation'
fused_act_arg.s = activation_name_map[activation_op.type]
final_op = activation_op
final_op.output_shape = output_shape
self.resolved_ops.add(activation_op.name)
op_def.output.extend([final_op.name + ':0'])
self.add_output_shape(op_def, output_shape) self.add_output_shape(op_def, output_shape)
self.net_def.op.extend([op_def]) self.net_def.op.extend([op_def])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册