From 4a91acb3f6eaf34c51a49c0bea09285f93766c27 Mon Sep 17 00:00:00 2001
From: Niels Rogge <niels.rogge1@gmail.com>
Date: Thu, 17 Dec 2020 15:36:16 +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 ef21e05..9e7e256 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": 12,
   "num_hidden_layers": 12,