funasr/bin/train.py
@@ -96,9 +96,11 @@ logging.info(f"Loading pretrained params from {p}") load_pretrained_model( model=model, init_param=p, path=p, ignore_init_mismatch=kwargs.get("ignore_init_mismatch", True), oss_bucket=kwargs.get("oss_bucket", None), scope_map=kwargs.get("scope_map", None), excludes=kwargs.get("excludes", None), ) else: initialize(model, kwargs.get("init", "kaiming_normal"))
