嘉渊 2 лет назад
Родитель
Сommit
9d8bb85a40
1 измененных файлов с 1 добавлено и 0 удалено
  1. 1 0
      funasr/bin/train.py

+ 1 - 0
funasr/bin/train.py

@@ -65,6 +65,7 @@ def get_parser():
     )
     parser.add_argument(
         "--dist_rank",
+        type=int,
         default=None,
         help="node rank for distributed training",
     )