update

嘉渊

2023-04-25 0237e3262522e47d6e47262403d47f4889c2f000

 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)