dtransposed commited on
Commit
286e55d
·
verified ·
1 Parent(s): 18d3a06

Update config.pbtxt

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