finetune

游雁

2024-03-24 ed952ff630a1ad89df488d3c657ef736129e6c10

 funasr/datasets/dataloader_entry.py

@@ -4,7 +4,7 @@

from funasr.register import tables

@tables.register("dataloader_classes", "DataloaderMapStyle")
# @tables.register("dataloader_classes", "DataloaderMapStyle")
def DataloaderMapStyle(frontend=None, tokenizer=None, **kwargs):
   # dataset
   logging.info("Build dataloader")
@@ -25,7 +25,7 @@
   
   return dataloader_tr, dataloader_val

# @tables.register("dataloader_classes", "DataloaderMapStyle")
@tables.register("dataloader_classes", "DataloaderMapStyle")
class DataloaderMapStyle:
   def __init__(self, frontend=None, tokenizer=None, **kwargs):
      # dataset