jilijeanlouis commited on
Commit
dfd07b6
1 Parent(s): 2675e5f

Upload config.pbtxt

Browse files
named-entity-recognition_dbmdz_bert-large-cased-finetuned-conll03-english_onnx_model/config.pbtxt ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: "named-entity-recognition_dbmdz_bert-large-cased-finetuned-conll03-english_onnx_model"
2
+ max_batch_size: 0
3
+ platform: "onnxruntime_onnx"
4
+ default_model_filename: "model.onnx"
5
+
6
+ input [
7
+ {
8
+ name: "input_ids"
9
+ data_type: TYPE_INT64
10
+ dims: [-1, -1]
11
+ },
12
+ {
13
+ name: "input.3"
14
+ data_type: TYPE_INT64
15
+ dims: [1, 6]
16
+ },
17
+ {
18
+ name: "attention_mask"
19
+ data_type: TYPE_INT64
20
+ dims: [-1, -1]
21
+ }
22
+ ]
23
+
24
+ output {
25
+ name: "logits"
26
+ data_type: TYPE_FP32
27
+ dims: [-1,-1]
28
+ }
29
+
30
+ instance_group [
31
+ {
32
+ count: 1
33
+ kind: KIND_GPU
34
+ }
35
+ ]