name: "named-entity-recognition_dbmdz_bert-large-cased-finetuned-conll03-english_onnx_model" | |
max_batch_size: 0 | |
platform: "onnxruntime_onnx" | |
default_model_filename: "model.onnx" | |
input [ | |
{ | |
name: "input_ids" | |
data_type: TYPE_INT64 | |
dims: [-1, -1] | |
}, | |
{ | |
name: "input.3" | |
data_type: TYPE_INT64 | |
dims: [1, 6] | |
}, | |
{ | |
name: "attention_mask" | |
data_type: TYPE_INT64 | |
dims: [-1, -1] | |
} | |
] | |
output { | |
name: "logits" | |
data_type: TYPE_FP32 | |
dims: [-1,-1] | |
} | |
instance_group [ | |
{ | |
count: 1 | |
kind: KIND_GPU | |
} | |
] |