test

语帆

2024-02-28 52fee96d71ba96fd09ad453dbae1926a1d601a56

test

 funasr/auto/auto_model.py

@@ -141,7 +141,7 @@
            kwargs = download_model(**kwargs)
        
        set_all_random_seed(kwargs.get("seed", 0))
        pdb.set_trace()

        device = kwargs.get("device", "cuda")
        if not torch.cuda.is_available() or kwargs.get("ngpu", 1) == 0:
            device = "cpu"

 funasr/models/lcbnet/model.py

@@ -274,15 +274,12 @@
                ind: int
        """
        with autocast(False):
            pdb.set_trace()
            # Data augmentation
            if self.specaug is not None and self.training:
                speech, speech_lengths = self.specaug(speech, speech_lengths)
            pdb.set_trace()
            # Normalization for feature: e.g. Global-CMVN, Utterance-CMVN
            if self.normalize is not None:
                speech, speech_lengths = self.normalize(speech, speech_lengths)
        pdb.set_trace()
        # Forward encoder
        # feats: (Batch, Length, Dim)
        # -> encoder_out: (Batch, Length2, Dim2)