Update classification/modeling_bbb.py
Browse files
classification/modeling_bbb.py
CHANGED
|
@@ -49,6 +49,8 @@ class BBBModelForSequenceClassification(PreTrainedModel):
|
|
| 49 |
nn.Linear(config.proj_dim, 1)
|
| 50 |
)
|
| 51 |
|
|
|
|
|
|
|
| 52 |
def _init_weights(self, module):
|
| 53 |
if isinstance(module, nn.Linear):
|
| 54 |
module.weight.data.normal_(mean=0.0, std=1.0)
|
|
|
|
| 49 |
nn.Linear(config.proj_dim, 1)
|
| 50 |
)
|
| 51 |
|
| 52 |
+
self.post_init()
|
| 53 |
+
|
| 54 |
def _init_weights(self, module):
|
| 55 |
if isinstance(module, nn.Linear):
|
| 56 |
module.weight.data.normal_(mean=0.0, std=1.0)
|