wav2vec2_en, test=asr

3 years ago · 0f766848c2
parent a0e862d288
commit 0f766848c2
1 changed files with 1 additions and 1 deletions
--- a/paddlespeech/s2t/exps/wav2vec2/model.py
+++ b/paddlespeech/s2t/exps/wav2vec2/model.py
@ -279,7 +279,7 @@ class Wav2Vec2ASRTrainer(Trainer):
        logger.info("Setup model!")

        # setup speech augmentation for wav2vec2
-        if hasattr(config, 'speech_augment'):
+        if hasattr(config, 'speech_augment') and self.train:
            speechaugment_config = TimeDomainSpecAugmentConfig(
                config.speech_augment)
            self.speech_augmentation = TimeDomainSpecAugment(