游雁
2024-02-22 4a71f20682f3d9a665cfb67afdaf673cc81d81b1
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"))