Fix TAPAS doc

This commit is contained in:
Lysandre 2020-12-17 11:24:23 -05:00
parent 6d2e864db7
commit e0790cca78

View File

@ -346,7 +346,7 @@ of that:
... inputs,
... outputs.logits.detach(),
... outputs.logits_aggregation.detach()
...)
... )
>>> # let's print out the results:
>>> id2aggregation = {0: "NONE", 1: "SUM", 2: "AVERAGE", 3:"COUNT"}