Update modeling_textcnn.py
Browse files- modeling_textcnn.py +2 -2
modeling_textcnn.py
CHANGED
|
@@ -39,11 +39,11 @@ class TextCNNPreTrainedModel(PreTrainedModel):
|
|
| 39 |
"attention_mask": input_ids.ne(pad_token),
|
| 40 |
"input_ids": input_ids,
|
| 41 |
}
|
| 42 |
-
dummy_inputs
|
| 43 |
|
| 44 |
|
| 45 |
class TextCNNModel(TextCNNPreTrainedModel):
|
| 46 |
-
""" A
|
| 47 |
|
| 48 |
def __init__(self, config):
|
| 49 |
super().__init__(config)
|
|
|
|
| 39 |
"attention_mask": input_ids.ne(pad_token),
|
| 40 |
"input_ids": input_ids,
|
| 41 |
}
|
| 42 |
+
return dummy_inputs
|
| 43 |
|
| 44 |
|
| 45 |
class TextCNNModel(TextCNNPreTrainedModel):
|
| 46 |
+
""" A style classifier Text-CNN """
|
| 47 |
|
| 48 |
def __init__(self, config):
|
| 49 |
super().__init__(config)
|