Update modeling_norbert.py

modeling_norbert.py

@@ -415,9 +415,9 @@ class Classifier(nn.Module):
             nn.Linear(config.hidden_size, num_labels)
         )
         self.hidden_size = config.hidden_size
-        self.initialize()
+        self._init_weights()
 
-    def initialize(self):
+    def _init_weights(self):
         std = math.sqrt(2.0 / (5.0 * self.hidden_size))
         nn.init.trunc_normal_(self.nonlinearity[1].weight, mean=0.0, std=std, a=-2*std, b=2*std)
         nn.init.trunc_normal_(self.nonlinearity[-1].weight, mean=0.0, std=std, a=-2*std, b=2*std)