[fix] fix spec aug training bug (#377)

wenet-e2e · Oct 22, 2024 · e9bbf73 · e9bbf73
1 parent 2751dd2
commit e9bbf73
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/wespeaker/utils/executor.py b/wespeaker/utils/executor.py
@@ -54,7 +54,8 @@ def run_epoch(dataloader, epoch_iter, model, criterion, optimizer, scheduler,
                     features, **configs['dataset_args'].get('cmvn_args', {}))
             # spec augmentation
             if configs['dataset_args'].get('spec_aug', False):
-                features = spec_aug(features, **configs['spec_aug_args'])
+                features = spec_aug(features,
+                                    **configs['dataset_args']['spec_aug_args'])
 
         with torch.cuda.amp.autocast(enabled=configs['enable_amp']):
             outputs = model(features)  # (embed_a,embed_b) in most cases