This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
Register
Sign In
hummingbird
/
fastNLP
Not watched
Unwatch
Watch all
Watch but not notify
2
Star
0
Fork
0
Code
Releases
13
Wiki
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
Browse Source
修复寻找hyper-parameter数量的bug
tags/v0.5.5
yhcc
GitHub
5 years ago
parent
4ceca47352
commit
3761c32beb
No known key found for this signature in database
GPG Key ID:
4AEE18F83AFDEB23
1 changed files
with
1 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+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)
Write
Preview
Loading…
Cancel
Save