Add uint8 support in data_feeder.py

7 years ago · 9752e5bcbd
parent 25aa45394b
commit 9752e5bcbd
1 changed files with 3 additions and 1 deletions
--- a/python/paddle/fluid/data_feeder.py
+++ b/python/paddle/fluid/data_feeder.py
@ -36,9 +36,11 @@ class DataToLoDTensorConverter(object):
            self.dtype = 'float64'
        elif dtype == core.VarDesc.VarType.INT32:
            self.dtype = 'int32'
+        elif dtype == core.VarDesc.VarType.UINT8:
+            self.dtype = 'uint8'
        else:
            raise ValueError("dtype must be any of [int32, float32, int64, "
-                             "float64]")
+                             "float64, uint8]")

        self.data = []
        self.lod = []