Add _init_weights to model

model.py

@@ -56,3 +56,6 @@ class BertMesh(PreTrainedModel):
             outs = self.dropout_layer(outs)
             outs = torch.sigmoid(self.linear_out(outs))
         return outs
+
+    def _init_weights(self, module):
+        pass