From 0f97ff0368a856ecc9f7904b43c45f64399ab23e Mon Sep 17 00:00:00 2001
From: yukavio <67678385+yukavio@users.noreply.github.com>
Date: Tue, 22 Dec 2020 13:59:34 +0800
Subject: [PATCH] fix flops (#29818)

---
 python/paddle/hapi/dynamic_flops.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/paddle/hapi/dynamic_flops.py b/python/paddle/hapi/dynamic_flops.py
index 8f6697872ce..bfbb483ac31 100644
--- a/python/paddle/hapi/dynamic_flops.py
+++ b/python/paddle/hapi/dynamic_flops.py
@@ -255,7 +255,8 @@ def dynamic_flops(model, inputs, custom_ops=None, print_detail=False):
     for m in model.sublayers():
         if len(list(m.children())) > 0:
             continue
-        if hasattr(m, 'total_ops') and hasattr(m, 'total_params'):
+        if set(['total_ops', 'total_params', 'input_shape',
+                'output_shape']).issubset(set(list(m._buffers.keys()))):
             total_ops += m.total_ops
             total_params += m.total_params
 
-- 
GitLab