diff --git a/python/pipeline/channel.py b/python/pipeline/channel.py index a00d8e3a62bd2a332d37fb891c578b2afb0ff901..b7c229ea6f85794849e156524616647f4bc121ec 100644 --- a/python/pipeline/channel.py +++ b/python/pipeline/channel.py @@ -127,7 +127,6 @@ class ChannelData(object): def get_size(self): size = 0 - dict_data = None if isinstance(self.dictdata, dict): for k in self.dictdata: size += sys.getsizeof(self.dictdata[k]) + sys.getsizeof(k) diff --git a/python/pipeline/pipeline_client.py b/python/pipeline/pipeline_client.py index 5585463e8edd859d9e542d22d7bed0da53e7d857..cff7e9c3f77db88c762383b83dc466ed2a8240ae 100644 --- a/python/pipeline/pipeline_client.py +++ b/python/pipeline/pipeline_client.py @@ -108,7 +108,9 @@ class PipelineClient(object): if isinstance(value, np.ndarray): # copy shape - _LOGGER.info("value shape is {}".format(value.shape)) + _LOGGER.debug( + "key:{}, use_tensor_bytes:{}, value.shape:{}, value.dtype:{}". + format(key, use_tensor_bytes, value.shape, value.dtype)) for one_dim in value.shape: one_tensor.shape.append(one_dim) @@ -118,6 +120,7 @@ class PipelineClient(object): np.save(np_bytes, value, allow_pickle=True) one_tensor.byte_data = np_bytes.getvalue() one_tensor.elem_type = 13 #13 => bytes in proto + continue flat_value = value.flatten().tolist() # copy data