FUNASR训练
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
游雁
2024-03-18 3d73f50beb39ca3bb0ac98a6b052fcfdb3be1f1d 
 funasr/datasets/audio_datasets/datasets.py


@@ -90,7 +90,7 @@




        for key, data_list in outputs.items():


            if isinstance(data_list[0], torch.Tensor):


                if data_list[0].dtype == torch.int64:


                if data_list[0].dtype == torch.int64 or data_list[0].dtype == torch.int32:


    


                    pad_value = self.int_pad_value


                else:


@@ -192,7 +192,7 @@




        for key, data_list in outputs.items():


            if isinstance(data_list[0], torch.Tensor):


                if data_list[0].dtype == torch.int64:


                if data_list[0].dtype == torch.int64 or data_list[0].dtype == torch.int32:


                    pad_value = self.int_pad_value


                else:


                    pad_value = self.float_pad_value