changdazhou 1 سال پیش
والد
کامیت
42c41ac3bd
1فایلهای تغییر یافته به همراه1 افزوده شده و 1 حذف شده
  1. 1 1
      paddlex/repo_apis/base/runner.py

+ 1 - 1
paddlex/repo_apis/base/runner.py

@@ -184,7 +184,7 @@ class BaseRunner(metaclass=abc.ABCMeta):
         if device is None:
             return args, None
         device, dev_ids = parse_device(device)
-        if len(dev_ids) == 0:
+        if dev_ids is None or len(dev_ids) == 0:
             return args, None
         else:
             num_devices = len(dev_ids)