提交 7c73360f 编写于 作者: L LielinJiang 提交者: whs

fix astar2019 flop count bug (#3197)

上级 da001769
...@@ -116,7 +116,7 @@ def _summary_model(block_vars, one_op): ...@@ -116,7 +116,7 @@ def _summary_model(block_vars, one_op):
out_data_shape = block_vars[one_op.output("Output")[0]].shape out_data_shape = block_vars[one_op.output("Output")[0]].shape
c_out, c_in, k_h, k_w = k_arg_shape c_out, c_in, k_h, k_w = k_arg_shape
_, c_out_, h_out, w_out = out_data_shape _, c_out_, h_out, w_out = out_data_shape
assert c_out == c_out_, 'shape error!' #assert c_out == c_out_, 'shape error!'
k_groups = one_op.attr("groups") k_groups = one_op.attr("groups")
kernel_ops = k_h * k_w * (c_in / k_groups) kernel_ops = k_h * k_w * (c_in / k_groups)
bias_ops = 0 if one_op.input("Bias") == [] else 1 bias_ops = 0 if one_op.input("Bias") == [] else 1
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册