嘉渊
2023-04-24 0eb3b1161b8c832a3a54831d6879b4923f23b265
funasr/bin/train.py
@@ -444,7 +444,7 @@
    # ddp init
    os.environ['CUDA_VISIBLE_DEVICES'] = str(args.gpu_id)
    args.distributed = args.dist_world_size > 1
    args.distributed = args.ngpu > 1 or args.dist_world_size > 1
    distributed_option = build_distributed(args)
    # for logging