Add 2 parameters for pipeline

This commit is contained in:
Niels Rogge 2020-12-18 09:00:02 +00:00 committed by huggingface-web
parent f46df6b4fd
commit a542e5efa3
1 changed files with 7 additions and 0 deletions

View File

@ -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,