FUNASR训练
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
游雁
2024-03-22 fcbbe8af9f22a41611d9506af17cae1e422f9fec 
 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