funasr1.0

游雁

2023-12-28 c9c95ec30aba4e114003d5196c06bb8ae5ca8cc4

 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")))