Commit 39b010ab authored by Yuxin Wu's avatar Yuxin Wu

fix import (fix #483)

parent a6b091aa
......@@ -37,7 +37,7 @@ class MultiProcessPredictWorker(multiprocessing.Process):
have workers that run on multiGPUs
"""
if self.idx != 0:
from tensorpack.models.common import disable_layer_logging
from tensorpack.models.registry import disable_layer_logging
disable_layer_logging()
self.predictor = OfflinePredictor(self.config)
if self.idx == 0:
......
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