julien-c HF staff commited on
Commit
af07a3f
1 Parent(s): a81c004

serialize some info as yaml

Browse files
Files changed (2) hide show
  1. config.yml +6 -0
  2. hparams.yml +29 -0
config.yml ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ model:
2
+ class: PyanNet
3
+ module: pyannote.audio.models.segmentation.PyanNet
4
+ versions:
5
+ pyannote.audio: 0.0.1
6
+ torch: 1.7.0
hparams.yml ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ learning_rate: 0.0001
2
+ linear:
3
+ hidden_size: 128
4
+ num_layers: 2
5
+ lstm:
6
+ batch_first: true
7
+ bidirectional: true
8
+ hidden_size: 128
9
+ num_layers: 4
10
+ model_introspection: !!python/object:pyannote.audio.core.model.ModelIntrospection
11
+ dimension: 1
12
+ inc_num_frames: 1
13
+ inc_num_samples: 216
14
+ min_num_frames: 2
15
+ min_num_samples: 1059
16
+ num_channels: 1
17
+ sample_rate: 16000
18
+ sincnet:
19
+ sample_rate: 16000
20
+ stride: 8
21
+ task_specifications: !!python/object:pyannote.audio.core.task.TaskSpecification
22
+ classes:
23
+ - speech
24
+ duration: 2.0
25
+ permutation_invariant: false
26
+ problem: !!python/object/apply:pyannote.audio.core.task.Problem
27
+ - 0
28
+ scale: !!python/object/apply:pyannote.audio.core.task.Scale
29
+ - 1