Browse Source

修复寻找hyper-parameter数量的bug

tags/v0.5.5
yhcc GitHub 5 years ago
parent
commit
3761c32beb
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      fastNLP/modules/utils.py

+ 1
- 1
fastNLP/modules/utils.py View File

@@ -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)


Loading…
Cancel
Save