From 5dae0ee497edf843017aa227a5253165cb066c2d Mon Sep 17 00:00:00 2001 From: xuyige Date: Sat, 3 Nov 2018 01:52:41 +0800 Subject: [PATCH] fix a bug in initial_method function --- 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 12efe1c8..5f5503bb 100644 --- a/fastNLP/modules/utils.py +++ b/fastNLP/modules/utils.py @@ -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_