diff --git a/funasr/tasks/abs_task.py b/funasr/tasks/abs_task.py index 14987f1e4..775cba86a 100644 --- a/funasr/tasks/abs_task.py +++ b/funasr/tasks/abs_task.py @@ -464,6 +464,12 @@ class AbsTask(ABC): default=sys.maxsize, help="The maximum number update step to train", ) + parser.add_argument( + "--batch_interval", + type=int, + default=10000, + help="The batch interval for saving model.", + ) group.add_argument( "--patience", type=int_or_none, @@ -1576,13 +1582,18 @@ class AbsTask(ABC): ) -> AbsIterFactory: assert check_argument_types() + if args.frontend_conf is not None and "fs" in args.frontend_conf: + dest_sample_rate = args.frontend_conf["fs"] + else: + dest_sample_rate = 16000 + dataset = ESPnetDataset( iter_options.data_path_and_name_and_type, float_dtype=args.train_dtype, preprocess=iter_options.preprocess_fn, max_cache_size=iter_options.max_cache_size, max_cache_fd=iter_options.max_cache_fd, - dest_sample_rate=args.frontend_conf["fs"], + dest_sample_rate=dest_sample_rate, ) cls.check_task_requirements( dataset, args.allow_variable_data_keys, train=iter_options.train diff --git a/funasr/tasks/asr.py b/funasr/tasks/asr.py index 6e0f16acf..e15147332 100644 --- a/funasr/tasks/asr.py +++ b/funasr/tasks/asr.py @@ -412,12 +412,6 @@ class ASRTask(AbsTask): default="13_15", help="The range of noise decibel level.", ) - parser.add_argument( - "--batch_interval", - type=int, - default=10000, - help="The batch interval for saving model.", - ) for class_choices in cls.class_choices_list: # Append -- and --_conf.