shixian.shi пре 2 година
родитељ
комит
1d1a33d5be
1 измењених фајлова са 1 додато и 0 уклоњено
  1. 1 0
      funasr/tasks/abs_task.py

+ 1 - 0
funasr/tasks/abs_task.py

@@ -1150,6 +1150,7 @@ class AbsTask(ABC):
     def main_worker(cls, args: argparse.Namespace):
         assert check_argument_types()
 
+        args.ngpu = 0
         # 0. Init distributed process
         distributed_option = build_dataclass(DistributedOption, args)
         # Setting distributed_option.dist_rank, etc.