From 818fd71fbac74b401a14925f12d95f216f4f4a28 Mon Sep 17 00:00:00 2001 From: xuyige Date: Sat, 3 Nov 2018 02:22:13 +0800 Subject: [PATCH] fix a bug in initial methods --- fastNLP/modules/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fastNLP/modules/utils.py b/fastNLP/modules/utils.py index 2eaff11c..21497037 100644 --- a/fastNLP/modules/utils.py +++ b/fastNLP/modules/utils.py @@ -32,9 +32,9 @@ def initial_parameter(net, initial_method=None): elif initial_method == 'xavier_normal': init_method = init.xavier_normal_ elif initial_method == 'kaiming_normal' or initial_method == 'msra': - init_method = init.kaiming_normal + init_method = init.kaiming_normal_ elif initial_method == 'kaiming_uniform': - init_method = init.kaiming_uniform + init_method = init.kaiming_uniform_ elif initial_method == 'orthogonal': init_method = init.orthogonal_ elif initial_method == 'sparse':