Browse Source

fix a bug in initial_method function

tags/v0.2.0
xuyige 6 years ago
parent
commit
5dae0ee497
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      fastNLP/modules/utils.py

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

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



Loading…
Cancel
Save