File tree Expand file tree Collapse file tree 1 file changed +18
-0
lines changed Expand file tree Collapse file tree 1 file changed +18
-0
lines changed Original file line number Diff line number Diff line change @@ -71,6 +71,12 @@ def set_default_initializer(self, initializer):
71
71
"""
72
72
Set the default initializer, the initializer should be Constant,
73
73
Uniform, Normal, Xavier, MSRA.
74
+
75
+ Args:
76
+ initializer(Initializer): the initializer to set.
77
+
78
+ Returns:
79
+ None
74
80
"""
75
81
if initializer is None :
76
82
if self .initializer is None :
@@ -85,12 +91,24 @@ def set_default_initializer(self, initializer):
85
91
def set_default_param_initializer (self ):
86
92
"""
87
93
Set the default initializer for the parameter with Xavier.
94
+
95
+ Args:
96
+ None.
97
+
98
+ Returns:
99
+ None.
88
100
"""
89
101
self .set_default_initializer (Xavier ())
90
102
91
103
def set_default_bias_initializer (self ):
92
104
"""
93
105
Set the default initializer for the bias with Constant(0.0).
106
+
107
+ Args:
108
+ None.
109
+
110
+ Returns:
111
+ None.
94
112
"""
95
113
self .set_default_initializer (Constant (0.0 ))
96
114
You can’t perform that action at this time.
0 commit comments