fix a bug in initial_method function
@@ -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_
