游雁
2024-01-12 247c763286c780e3f1a849148528b12e15506721
funasr/bin/train.py
@@ -23,9 +23,11 @@
from funasr.download.download_from_hub import download_model
from funasr.register import tables
@hydra.main(config_name=None, version_base=None)
@hydra.main(config_name=None)
def main_hydra(kwargs: DictConfig):
   import pdb; pdb.set_trace()
   if kwargs.get("debug", False):
      import pdb; pdb.set_trace()
   assert "model" in kwargs
   if "model_conf" not in kwargs:
      logging.info("download models from model hub: {}".format(kwargs.get("model_hub", "ms")))