嘉渊 2 anos atrás
pai
commit
f16c6c2926
1 arquivos alterados com 0 adições e 2 exclusões
  1. 0 2
      funasr/bin/build_trainer.py

+ 0 - 2
funasr/bin/build_trainer.py

@@ -627,8 +627,6 @@ def build_trainer(modelscope_dict,
     torch.backends.cudnn.deterministic = args.cudnn_deterministic
 
     # ddp init
-    os.environ['CUDA_VISIBLE_DEVICES'] = str(args.gpu_id)
-    args.distributed = args.ngpu > 1 or args.dist_world_size > 1
     distributed_option = build_distributed(args)
 
     # for logging