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_