From 3761c32bebae2d67a293da90c57f8a288ff15d3f Mon Sep 17 00:00:00 2001 From: yhcc Date: Wed, 30 Oct 2019 17:06:56 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=AF=BB=E6=89=BEhyper-param?= =?UTF-8?q?eter=E6=95=B0=E9=87=8F=E7=9A=84bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- fastNLP/modules/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)