From 2a5bd711312bba91e588a3f3283a1379ddbe9b12 Mon Sep 17 00:00:00 2001 From: xuyige Date: Sat, 3 Nov 2018 02:14:13 +0800 Subject: [PATCH] fix a bug in initial methods --- fastNLP/modules/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fastNLP/modules/utils.py b/fastNLP/modules/utils.py index 5f5503bb..2eaff11c 100644 --- a/fastNLP/modules/utils.py +++ b/fastNLP/modules/utils.py @@ -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':