funasr1.0 finetune

游雁

2024-01-16 2ed3f46f40cd5da19cad76a97b52c46b2869d5ed

 funasr/bin/train.py

@@ -40,8 +40,7 @@


def main(**kwargs):
    # preprocess_config(kwargs)
    # import pdb; pdb.set_trace()
    
    # set random seed
    tables.print()
    set_all_random_seed(kwargs.get("seed", 0))
@@ -169,6 +168,8 @@
        local_rank=local_rank,
        use_ddp=use_ddp,
        use_fsdp=use_fsdp,
        output_dir=kwargs.get("output_dir", "./exp"),
        resume=kwargs.get("resume", True),
        **kwargs.get("train_conf"),
    )
    trainer.run()