FUNASR训练
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
gaochangfeng
2024-04-10 851e3e3ef83d0769d9bde172d8841f6b20e3e377 
 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