fp16

游雁

2024-02-23 8827e26b8d487f123f8d7d5cbd8d00b81dcefcff

 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