aky15
2023-05-24 8c24f52fd25fce2df46dc4d9dffb45619dc38f9f
funasr/datasets/large_datasets/build_dataloader.py
@@ -73,8 +73,8 @@
            seg_dict = load_seg_dict(args.seg_dict_file)
        if hasattr(args, "punc_dict_file") and args.punc_dict_file is not None:
            punc_dict = read_symbol_table(args.punc_dict_file)
        if hasattr(args, "bpemodel_file") and args.bpemodel_file is not None:
            bpe_tokenizer = SentencepiecesTokenizer(args.bpemodel_file)
        if hasattr(args, "bpemodel") and args.bpemodel is not None:
            bpe_tokenizer = SentencepiecesTokenizer(args.bpemodel)
        self.dataset_conf = args.dataset_conf
        self.frontend_conf = args.frontend_conf
        logging.info("dataloader config: {}".format(self.dataset_conf))