jackstanley
/

HAT-base-4096-modified

hierarchical-transformer

Model card Files Files and versions

jackstanley commited on Nov 18, 2022

Commit

9cee461

·

1 Parent(s): 7140f4d

Update modelling_hat.py

Files changed (1) hide show

modelling_hat.py +1 -1

modelling_hat.py CHANGED Viewed

@@ -1926,7 +1926,7 @@ class HATForSequenceClassification(HATPreTrainedModel):
             loss=loss,
             logits=logits,
             hidden_states=outputs.hidden_states,
-            attentions=outputs.attentions,
         )

             loss=loss,
             logits=logits,
             hidden_states=outputs.hidden_states,
+            attentions=(outputs.attentions, outputs.sentence_attentions),
         )