use HookedSession to do inference. fix #161

tensorpack/callbacks/inference.py

@@ -91,8 +91,6 @@ class ScalarStats(Inferencer):
         self.stats = []
 
     def _datapoint(self, output):
-        for o in output:
-            assert isinstance(o, (float, np.float32)), type(o)
         self.stats.append(output)
 
     def _after_inference(self):

tensorpack/predict/base.py

@@ -194,6 +194,10 @@ class PredictorTowerBuilder(object):
 
     @staticmethod
     def get_tensors_maybe_in_tower(placeholder_names, names, k, prefix=''):
+        """
+        Args:
+            placeholders (list): A list of __op__ name.
+        """
         def maybe_inside_tower(name):
             name = get_op_tensor_name(name)[0]
             if name in placeholder_names: