diff --git a/PaddleCV/Research/astar2019/utils.py b/PaddleCV/Research/astar2019/utils.py index ed5c51e2da2781ff59ef68d1ef56dc353b147432..0ed49b81e6482c21fd88fc0988fa5b90d112417d 100644 --- a/PaddleCV/Research/astar2019/utils.py +++ b/PaddleCV/Research/astar2019/utils.py @@ -116,7 +116,7 @@ def _summary_model(block_vars, one_op): out_data_shape = block_vars[one_op.output("Output")[0]].shape c_out, c_in, k_h, k_w = k_arg_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") kernel_ops = k_h * k_w * (c_in / k_groups) bias_ops = 0 if one_op.input("Bias") == [] else 1