Commit ef24ea95 authored by Yuxin Wu's avatar Yuxin Wu

fix lint

parent 4a1af743
......@@ -114,8 +114,8 @@ class MultiProcessDatasetPredictor(DatasetPredictorBase):
except KeyError:
gpus = list(range(get_num_gpu()))
assert len(gpus) >= self.nr_proc, \
"nr_proc={} while only {} gpus available".format(
self.nr_proc, len(gpus))
"nr_proc={} while only {} gpus available".format(
self.nr_proc, len(gpus))
else:
gpus = ['-1'] * self.nr_proc
# worker produces (idx, result) to outqueue
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment