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_method function

Browse Source
This commit is contained in:
xuyige 2018-11-03 01:52:41 +08:00
parent 19332b8b08
commit 5dae0ee497
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

@ -42,7 +42,7 @@ def initial_parameter(net, initial_method=None):
elif initial_method == 'normal': elif initial_method == 'normal':
init_method = init.normal_ init_method = init.normal_
elif initial_method == 'uniform': elif initial_method == 'uniform':
initial_method = init.uniform_ init_method = init.uniform_
else: else:
init_method = init.xavier_normal_ init_method = init.xavier_normal_