funasr/bin/train.py
@@ -59,6 +59,7 @@ ) parser.add_argument( "--dist_world_size", type=int, default=1, help="number of nodes for distributed training", ) @@ -69,6 +70,7 @@ ) parser.add_argument( "--local_rank", type=int, default=None, help="local rank for distributed training", ) @@ -465,7 +467,6 @@ args, extra_task_params = parser.parse_known_args() if extra_task_params: args = build_args(args, parser, extra_task_params) # args = argparse.Namespace(**vars(args), **vars(task_args)) # set random seed set_all_random_seed(args.seed)