Update modeling_norbert.py

modeling_norbert.py

@@ -440,7 +440,10 @@ class NorbertForSequenceClassification(NorbertModel):
         self.head = Classifier(config, self.num_labels)
 
     def post_init(self):
-        self.head.initialize()
+        self.head._init_weights()
+
+    def _init_weights(self):
+        self.head._init_weights()
 
     def forward(
         self,
@@ -509,7 +512,7 @@ class NorbertForTokenClassification(NorbertModel):
         self.head = Classifier(config, self.num_labels)
 
     def post_init(self):
-        self.head.initialize()
+        self.head._init_weights()
 
     def forward(
         self,