dtransposed commited on
Commit
4bf5aeb
·
verified ·
1 Parent(s): 84b359c

Update config.pbtxt

Browse files
Files changed (1) hide show
  1. config.pbtxt +32 -25
config.pbtxt CHANGED
@@ -1,29 +1,36 @@
1
  name: "voicemail_classifier"
2
  backend: "onnxruntime"
3
  max_batch_size: 64
4
- input {
5
- name: "audio_features"
6
- data_type: TYPE_FP32
7
- dims: 80
8
- dims: 3000
9
- }
10
- output {
11
- name: "logits"
12
- data_type: TYPE_FP32
13
- dims: 2
14
- }
15
- output {
16
- name: "probabilities"
17
- data_type: TYPE_FP32
18
- dims: 2
19
- }
20
- output {
21
- name: "predicted_class"
22
- data_type: TYPE_INT64
23
- dims: 1
24
- }
25
- instance_group {
26
- count: 1
27
- kind: KIND_CPU
28
- }
29
  ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  name: "voicemail_classifier"
2
  backend: "onnxruntime"
3
  max_batch_size: 64
4
+
5
+ input [
6
+ {
7
+ name: "audio_features"
8
+ data_type: TYPE_FP32
9
+ dims: [80, 3000]
10
+ }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  ]
12
+
13
+ output [
14
+ {
15
+ name: "logits"
16
+ data_type: TYPE_FP32
17
+ dims: [2]
18
+ },
19
+ {
20
+ name: "probabilities"
21
+ data_type: TYPE_FP32
22
+ dims: [2]
23
+ },
24
+ {
25
+ name: "predicted_class"
26
+ data_type: TYPE_INT64
27
+ dims: [1]
28
+ }
29
+ ]
30
+
31
+ instance_group [
32
+ {
33
+ count: 1
34
+ kind: KIND_CPU
35
+ }
36
+ ]