Update onnx_config.pbtxt
Browse files- onnx_config.pbtxt +3 -3
onnx_config.pbtxt
CHANGED
@@ -5,19 +5,19 @@ input [
|
|
5 |
{
|
6 |
name: "input_ids"
|
7 |
data_type: TYPE_INT64
|
8 |
-
dims: [ 512 ]
|
9 |
},
|
10 |
{
|
11 |
name: "attention_mask"
|
12 |
data_type: TYPE_INT64
|
13 |
-
dims: [ 512 ]
|
14 |
}
|
15 |
]
|
16 |
output [
|
17 |
{
|
18 |
name: "logits"
|
19 |
data_type: TYPE_FP32
|
20 |
-
dims: [ 2 ]
|
21 |
}
|
22 |
]
|
23 |
instance_group [
|
|
|
5 |
{
|
6 |
name: "input_ids"
|
7 |
data_type: TYPE_INT64
|
8 |
+
dims: [ -1, 512 ]
|
9 |
},
|
10 |
{
|
11 |
name: "attention_mask"
|
12 |
data_type: TYPE_INT64
|
13 |
+
dims: [ -1, 512 ]
|
14 |
}
|
15 |
]
|
16 |
output [
|
17 |
{
|
18 |
name: "logits"
|
19 |
data_type: TYPE_FP32
|
20 |
+
dims: [ -1, 2 ]
|
21 |
}
|
22 |
]
|
23 |
instance_group [
|