update

嘉渊

2023-04-25 7f7c23c36fab81f1f64e0ecdeb22875960b7200f

 funasr/bin/train.py

@@ -520,6 +520,10 @@
    prepare_data(args, distributed_option)

    model = build_model(args)
    model = model.to(
        dtype=getattr(torch, args.train_dtype),
        device="cuda" if args.ngpu > 0 else "cpu",
    )
    optimizers = build_optimizer(args, model=model)
    schedulers = build_scheduler(args, optimizers)