Nick Sorros commited on
Commit ·
44b4eae
1
Parent(s): 1b85e66
Add _init_weights to model
Browse files
model.py
CHANGED
|
@@ -56,3 +56,6 @@ class BertMesh(PreTrainedModel):
|
|
| 56 |
outs = self.dropout_layer(outs)
|
| 57 |
outs = torch.sigmoid(self.linear_out(outs))
|
| 58 |
return outs
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
outs = self.dropout_layer(outs)
|
| 57 |
outs = torch.sigmoid(self.linear_out(outs))
|
| 58 |
return outs
|
| 59 |
+
|
| 60 |
+
def _init_weights(self, module):
|
| 61 |
+
pass
|