funasr/bin/train.py
@@ -96,6 +96,7 @@ init_param = (init_param,) logging.info("init_param is not None: %s", init_param) for p in init_param: if os.path.exists(p): logging.info(f"Loading pretrained params from {p}") load_pretrained_model( model=model, @@ -106,6 +107,8 @@ excludes=kwargs.get("excludes", None), ) else: logging.info(f"Checkpoint does not exist, init randomly: {p}") else: initialize(model, kwargs.get("init", "kaiming_normal"))