From 7c73360fbaca386300376f9e63e2ac1701590cf2 Mon Sep 17 00:00:00 2001
From: LielinJiang <50691816+LielinJiang@users.noreply.github.com>
Date: Tue, 27 Aug 2019 14:13:17 +0800
Subject: [PATCH] fix astar2019 flop count bug (#3197)

---
 PaddleCV/Research/astar2019/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/PaddleCV/Research/astar2019/utils.py b/PaddleCV/Research/astar2019/utils.py
index ed5c51e2..0ed49b81 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
-- 
GitLab