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

fix a typo in dorefa

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