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