游雁
2024-02-23 ead9addaf7f4f779fac6c745125c31bf16578ff6
funasr/bin/train.py
@@ -108,8 +108,10 @@
                )
            else:
                logging.info(f"Checkpoint does not exist, init randomly: {p}")
    else:
    elif kwargs.get("init", None):
        initialize(model, kwargs.get("init", "kaiming_normal"))
    else:
        print("No initialize method")
    # freeze_param