hummingbird
/
fastNLP
Not watched
2
0
Fork 0
Code
Releases 13 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

fix a bug in initial methods

tags/v0.2.0
xuyige 6 years ago
parent
a6ecc8be83
commit
2a5bd71131
1 changed files with 1 additions and 1 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      fastNLP/modules/utils.py

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

@@ -34,7 +34,7 @@ def initial_parameter(net, initial_method=None):
elif initial_method == 'kaiming_normal' or initial_method == 'msra':
init_method = init.kaiming_normal
elif initial_method == 'kaiming_uniform':
init_method = init.kaiming_normal
init_method = init.kaiming_uniform
elif initial_method == 'orthogonal':
init_method = init.orthogonal_
elif initial_method == 'sparse':


Write Preview
Loading…
Cancel
Save