Add 2 parameters for pipeline
This commit is contained in:
parent
f46df6b4fd
commit
a542e5efa3
|
@ -1,4 +1,10 @@
|
||||||
{
|
{
|
||||||
|
"aggregation_labels": {
|
||||||
|
"0": "NONE",
|
||||||
|
"1": "SUM",
|
||||||
|
"2": "AVERAGE",
|
||||||
|
"3": "COUNT"
|
||||||
|
},
|
||||||
"aggregation_loss_weight": 1.0,
|
"aggregation_loss_weight": 1.0,
|
||||||
"aggregation_temperature": 1.0,
|
"aggregation_temperature": 1.0,
|
||||||
"allow_empty_column_selection": true,
|
"allow_empty_column_selection": true,
|
||||||
|
@ -25,6 +31,7 @@
|
||||||
"max_num_rows": 64,
|
"max_num_rows": 64,
|
||||||
"max_position_embeddings": 1024,
|
"max_position_embeddings": 1024,
|
||||||
"model_type": "tapas",
|
"model_type": "tapas",
|
||||||
|
"no_aggregation_label_index": 0,
|
||||||
"num_aggregation_labels": 4,
|
"num_aggregation_labels": 4,
|
||||||
"num_attention_heads": 12,
|
"num_attention_heads": 12,
|
||||||
"num_hidden_layers": 12,
|
"num_hidden_layers": 12,
|
||||||
|
|
Loading…
Reference in New Issue