experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Normalize sample rate of dev/test sets to avoid feature computation errors

Browse Source
This commit is contained in:
Reuben Morais 2021-04-06 11:41:12 +02:00
parent c78af058a5
commit 8cdaa18533
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
training/coqui_stt_training/train.py
View File

@ -438,6 +438,7 @@ def train():
dev_sets = [create_dataset([source], dev_sets = [create_dataset([source],
batch_size=FLAGS.dev_batch_size, batch_size=FLAGS.dev_batch_size,
train_phase=False, train_phase=False,
augmentations=[NormalizeSampleRate(FLAGS.audio_sample_rate)],
exception_box=exception_box, exception_box=exception_box,
process_ahead=len(Config.available_devices) * FLAGS.dev_batch_size * 2, process_ahead=len(Config.available_devices) * FLAGS.dev_batch_size * 2,
reverse=FLAGS.reverse_dev, reverse=FLAGS.reverse_dev,
@ -450,6 +451,7 @@ def train():
metrics_sets = [create_dataset([source], metrics_sets = [create_dataset([source],
batch_size=FLAGS.dev_batch_size, batch_size=FLAGS.dev_batch_size,
train_phase=False, train_phase=False,
augmentations=[NormalizeSampleRate(FLAGS.audio_sample_rate)],
exception_box=exception_box, exception_box=exception_box,
process_ahead=len(Config.available_devices) * FLAGS.dev_batch_size * 2, process_ahead=len(Config.available_devices) * FLAGS.dev_batch_size * 2,
reverse=FLAGS.reverse_dev, reverse=FLAGS.reverse_dev,