Commit 95b6437a authored by Yuxin Wu's avatar Yuxin Wu

fix a typo in dorefa

parent 734b64aa
...@@ -233,7 +233,7 @@ def get_config(): ...@@ -233,7 +233,7 @@ def get_config():
InferenceRunner(data_test, InferenceRunner(data_test,
[ScalarStats('cost'), [ScalarStats('cost'),
ClassificationError('wrong-top1', 'val-error-top1'), ClassificationError('wrong-top1', 'val-error-top1'),
ClassificationError('wrong-top5', 'val-error-top1')]) ClassificationError('wrong-top5', 'val-error-top5')])
]), ]),
model=Model(), model=Model(),
step_per_epoch=10000, step_per_epoch=10000,
......
...@@ -13,6 +13,7 @@ import multiprocessing ...@@ -13,6 +13,7 @@ import multiprocessing
import tensorflow as tf import tensorflow as tf
from tensorflow.contrib.layers import variance_scaling_initializer from tensorflow.contrib.layers import variance_scaling_initializer
from tensorpack import * from tensorpack import *
from tensorpack.utils.stat import RatioCounter
from tensorpack.tfutils.symbolic_functions import * from tensorpack.tfutils.symbolic_functions import *
from tensorpack.tfutils.summary import * from tensorpack.tfutils.summary import *
......
...@@ -87,7 +87,7 @@ class MultiProcessDatasetPredictor(DatasetPredictorBase): ...@@ -87,7 +87,7 @@ class MultiProcessDatasetPredictor(DatasetPredictorBase):
self.nr_proc, len(gpus)) self.nr_proc, len(gpus))
except KeyError: except KeyError:
# TODO number of GPUs not checked # TODO number of GPUs not checked
gpus = list(range(self.nr_gpu)) gpus = list(range(self.nr_proc))
else: else:
gpus = [''] * self.nr_proc gpus = [''] * self.nr_proc
self.workers = [MultiProcessQueuePredictWorker( self.workers = [MultiProcessQueuePredictWorker(
......
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