提交 f00e91f7 编写于 作者: S ShiningZhang

fix client.py: check if the feed of string_type is np.array

上级 f424ef1a
......@@ -417,7 +417,10 @@ class Client(object):
key)])
else:
string_lod_slot_batch.append([])
string_slot.append(feed_dict[key].tostring())
if type(feed_dict[key]) is np.ndarray:
string_slot.append(feed_dict[key].tostring())
else:
string_slot.append(feed_dict[key])
self.has_numpy_input = True
self.profile_.record('py_prepro_1')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册