funasr1.0

游雁

2023-12-27 ccb948895498b48e591f1f6a74cb62f4dcde8202

 funasr/bin/train.py

@@ -25,7 +25,9 @@

@hydra.main(config_name=None, version_base=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")))