From a6b6bab2c68177e2b1288a6c9c5c06626f1f6b5e Mon Sep 17 00:00:00 2001 From: Binbin Zhang Date: Tue, 22 Oct 2024 10:46:42 +0800 Subject: [PATCH] [fix] fix spec aug training bug --- wespeaker/utils/executor.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/wespeaker/utils/executor.py b/wespeaker/utils/executor.py index c9477ca5..5fd33b95 100644 --- 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