update

嘉渊

2023-04-25 74a2059e9c2d7985300758a7e3b2293014229523

 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)