Skip to content

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

fix import (fix #483)

parent a6b091aa
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/predict/concurrency.py
View file @ 39b010ab
...@@ -37,7 +37,7 @@ class MultiProcessPredictWorker(multiprocessing.Process): ...@@ -37,7 +37,7 @@ class MultiProcessPredictWorker(multiprocessing.Process):
have workers that run on multiGPUs have workers that run on multiGPUs
""" """
if self.idx != 0: if self.idx != 0:
from tensorpack.models.common import disable_layer_logging from tensorpack.models.registry import disable_layer_logging
disable_layer_logging() disable_layer_logging()
self.predictor = OfflinePredictor(self.config) self.predictor = OfflinePredictor(self.config)
if self.idx == 0: 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