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)
|