提交 9752e5bc 编写于 作者: F fengjiayi

Add uint8 support in data_feeder.py

上级 25aa4539
...@@ -36,9 +36,11 @@ class DataToLoDTensorConverter(object): ...@@ -36,9 +36,11 @@ class DataToLoDTensorConverter(object):
self.dtype = 'float64' self.dtype = 'float64'
elif dtype == core.VarDesc.VarType.INT32: elif dtype == core.VarDesc.VarType.INT32:
self.dtype = 'int32' self.dtype = 'int32'
elif dtype == core.VarDesc.VarType.UINT8:
self.dtype = 'uint8'
else: else:
raise ValueError("dtype must be any of [int32, float32, int64, " raise ValueError("dtype must be any of [int32, float32, int64, "
"float64]") "float64, uint8]")
self.data = [] self.data = []
self.lod = [] self.lod = []
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册