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 [
|