Commit
•
9cee461
1
Parent(s):
7140f4d
Update modelling_hat.py
Browse files- modelling_hat.py +1 -1
modelling_hat.py
CHANGED
@@ -1926,7 +1926,7 @@ class HATForSequenceClassification(HATPreTrainedModel):
|
|
1926 |
loss=loss,
|
1927 |
logits=logits,
|
1928 |
hidden_states=outputs.hidden_states,
|
1929 |
-
attentions=outputs.attentions,
|
1930 |
)
|
1931 |
|
1932 |
|
|
|
1926 |
loss=loss,
|
1927 |
logits=logits,
|
1928 |
hidden_states=outputs.hidden_states,
|
1929 |
+
attentions=(outputs.attentions, outputs.sentence_attentions),
|
1930 |
)
|
1931 |
|
1932 |
|