From a542e5efa3b64cc1f0619e9fc9be08118852cdcc Mon Sep 17 00:00:00 2001
From: Niels Rogge <niels.rogge1@gmail.com>
Date: Fri, 18 Dec 2020 09:00:02 +0000
Subject: [PATCH] Add 2 parameters for pipeline

---
 config.json | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/config.json b/config.json
index 9ef0302..b61c11b 100644
--- a/config.json
+++ b/config.json
@@ -1,4 +1,10 @@
 {
+  "aggregation_labels": {
+    "0": "NONE",
+    "1": "SUM",
+    "2": "AVERAGE",
+    "3": "COUNT"
+  },
   "aggregation_loss_weight": 1.0,
   "aggregation_temperature": 1.0,
   "allow_empty_column_selection": true,
@@ -25,6 +31,7 @@
   "max_num_rows": 64,
   "max_position_embeddings": 1024,
   "model_type": "tapas",
+  "no_aggregation_label_index": 0,
   "num_aggregation_labels": 4,
   "num_attention_heads": 12,
   "num_hidden_layers": 12,