From 75433126df2f6adfaf90c4a0b853ec37ed729892 Mon Sep 17 00:00:00 2001 From: LielinJiang <50691816+LielinJiang@users.noreply.github.com> Date: Mon, 15 Mar 2021 14:24:46 +0800 Subject: [PATCH] Fix summary bug when calaculating output shape (#31549) * fix summary bug --- python/paddle/hapi/model_summary.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/python/paddle/hapi/model_summary.py b/python/paddle/hapi/model_summary.py index babbe962a9..9f2769e1ca 100644 --- a/python/paddle/hapi/model_summary.py +++ b/python/paddle/hapi/model_summary.py @@ -341,10 +341,12 @@ def summary_string(model, input_size, dtypes=None): total_params += summary[layer]["nb_params"] try: - total_output += np.prod(summary[layer]["output_shape"]) + total_output += np.sum( + np.prod( + summary[layer]["output_shape"], axis=-1)) except: for output_shape in summary[layer]["output_shape"]: - total_output += np.prod(output_shape) + total_output += np.sum(np.prod(output_shape, axis=-1)) if "trainable" in summary[layer]: if summary[layer]["trainable"] == True: -- GitLab