test/fastNLP
1
0
Fork 0
mirror of https://gitee.com/fastnlp/fastNLP.git synced 2024-12-11 10:05:30 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix a bug in initial methods

Browse Source
This commit is contained in:
xuyige 2018-11-03 02:14:13 +08:00
parent a6ecc8be83
commit 2a5bd71131
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
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': elif initial_method == 'kaiming_normal' or initial_method == 'msra':
init_method = init.kaiming_normal init_method = init.kaiming_normal
elif initial_method == 'kaiming_uniform': elif initial_method == 'kaiming_uniform':
init_method = init.kaiming_normal init_method = init.kaiming_uniform
elif initial_method == 'orthogonal': elif initial_method == 'orthogonal':
init_method = init.orthogonal_ init_method = init.orthogonal_
elif initial_method == 'sparse': elif initial_method == 'sparse':