Skip to content

S
seminar-breakout
Commit ef24ea95 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix lint

parent 4a1af743
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tensorpack/predict/dataset.py
View file @ ef24ea95
...@@ -114,8 +114,8 @@ class MultiProcessDatasetPredictor(DatasetPredictorBase): ...@@ -114,8 +114,8 @@ class MultiProcessDatasetPredictor(DatasetPredictorBase):
except KeyError: except KeyError:
gpus = list(range(get_num_gpu())) gpus = list(range(get_num_gpu()))
assert len(gpus) >= self.nr_proc, \ assert len(gpus) >= self.nr_proc, \
"nr_proc={} while only {} gpus available".format( "nr_proc={} while only {} gpus available".format(
self.nr_proc, len(gpus)) self.nr_proc, len(gpus))
else: else:
gpus = ['-1'] * self.nr_proc gpus = ['-1'] * self.nr_proc
# worker produces (idx, result) to outqueue # 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