diff --git a/fastNLP/modules/utils.py b/fastNLP/modules/utils.py
index 54993479..18bef709 100644
--- a/fastNLP/modules/utils.py
+++ b/fastNLP/modules/utils.py
@@ -101,7 +101,7 @@ def summary(model: nn.Module):
             else:
                 nontrain.append(count_size(p.shape))
         for p in module.buffers():
-            buffer.append(count_size(p))
+            buffer.append(count_size(p.shape))
         for subm in module.children():
             layer_summary(subm)