Update executor.py

wenet-e2e · Nov 9, 2024 · 6119101 · 6119101
1 parent 9b5379e
commit 6119101
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/wespeaker/utils/executor.py b/wespeaker/utils/executor.py
@@ -57,7 +57,6 @@ def run_epoch(dataloader, epoch_iter, model, criterion, optimizer, scheduler,
                 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
             embeds = outputs[-1] if isinstance(outputs, tuple) else outputs
             outputs = model.module.projection(embeds, targets)