rename predictor_factory.py

tensorpack/callbacks/inference_runner.py

@@ -19,7 +19,7 @@ from ..dataflow.base import DataFlow
 
 from ..input_source import (
     InputSource, FeedInput, QueueInput)
-from ..graph_builder.predictor_factory import SimplePredictBuilder
+from ..graph_builder.predict import SimplePredictBuilder
 
 from .base import Callback
 from .group import Callbacks

tensorpack/predict/multigpu.py

@@ -5,7 +5,7 @@
 
 import tensorflow as tf
 from ..utils import logger
-from ..graph_builder.predictor_factory import SimplePredictBuilder
+from ..graph_builder.predict import SimplePredictBuilder
 from ..input_source import PlaceholderInput
 from .base import OnlinePredictor
 

tensorpack/train/tower.py

@@ -7,7 +7,7 @@ import six
 from abc import abstractmethod, ABCMeta
 
 from ..utils.argtools import call_only_once, memoized
-from ..graph_builder.predictor_factory import SimplePredictBuilder
+from ..graph_builder.predict import SimplePredictBuilder
 from ..input_source import PlaceholderInput
 from ..predict.base import OnlinePredictor
 

tensorpack/trainv1/base.py

@@ -21,7 +21,7 @@ from ..tfutils.sessinit import JustCurrentSession
 from ..tfutils.tower import TowerFuncWrapper
 
 from ..input_source import PlaceholderInput
-from ..graph_builder.predictor_factory import SimplePredictBuilder
+from ..graph_builder.predict import SimplePredictBuilder
 from ..predict.base import OnlinePredictor
 from ..callbacks.steps import MaintainStepCounter