fix flops (#29818)

5 years ago · 0f97ff0368
parent 01c37c8e02
commit 0f97ff0368
1 changed files with 2 additions and 1 deletions
--- 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