Upload folder using huggingface_hub
Browse files- modeling_chronogrid.py +4 -0
modeling_chronogrid.py
CHANGED
|
@@ -35,6 +35,10 @@ class ChronoGridModelForSequenceClassification(PreTrainedModel):
|
|
| 35 |
super().__init__(config)
|
| 36 |
self.num_classes = config.num_classes
|
| 37 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 38 |
# 1. Initialize your architecture
|
| 39 |
self.cnn = make_cnn_backbone(config.img_size)
|
| 40 |
self.bilstm = nn.LSTM(256, 128, num_layers=2, batch_first=True, bidirectional=True, dropout=0.3)
|
|
|
|
| 35 |
super().__init__(config)
|
| 36 |
self.num_classes = config.num_classes
|
| 37 |
|
| 38 |
+
# Bugfix for transformers newer versions requiring this attribute
|
| 39 |
+
if not hasattr(self, "all_tied_weights_keys"):
|
| 40 |
+
self.all_tied_weights_keys = {}
|
| 41 |
+
|
| 42 |
# 1. Initialize your architecture
|
| 43 |
self.cnn = make_cnn_backbone(config.img_size)
|
| 44 |
self.bilstm = nn.LSTM(256, 128, num_layers=2, batch_first=True, bidirectional=True, dropout=0.3)
|