julien-c's picture
julien-c HF staff
serialize some info as yaml
af07a3f
raw
history blame contribute delete
698 Bytes
learning_rate: 0.0001
linear:
hidden_size: 128
num_layers: 2
lstm:
batch_first: true
bidirectional: true
hidden_size: 128
num_layers: 4
model_introspection: !!python/object:pyannote.audio.core.model.ModelIntrospection
dimension: 1
inc_num_frames: 1
inc_num_samples: 216
min_num_frames: 2
min_num_samples: 1059
num_channels: 1
sample_rate: 16000
sincnet:
sample_rate: 16000
stride: 8
task_specifications: !!python/object:pyannote.audio.core.task.TaskSpecification
classes:
- speech
duration: 2.0
permutation_invariant: false
problem: !!python/object/apply:pyannote.audio.core.task.Problem
- 0
scale: !!python/object/apply:pyannote.audio.core.task.Scale
- 1