From 90941e412f2393d8d1787a53c6ff00712e609ffa Mon Sep 17 00:00:00 2001
From: Niels Rogge <niels.rogge1@gmail.com>
Date: Thu, 17 Dec 2020 16:34:13 +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 362f654..63a04a0 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": false,
@@ -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": 16,
   "num_hidden_layers": 24,