Skip to content

Commit 9752e5b

Browse files
committed
Add uint8 support in data_feeder.py
1 parent 25aa453 commit 9752e5b

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

python/paddle/fluid/data_feeder.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,9 +36,11 @@ def __init__(self, place, lod_level, shape, dtype):
3636
self.dtype = 'float64'
3737
elif dtype == core.VarDesc.VarType.INT32:
3838
self.dtype = 'int32'
39+
elif dtype == core.VarDesc.VarType.UINT8:
40+
self.dtype = 'uint8'
3941
else:
4042
raise ValueError("dtype must be any of [int32, float32, int64, "
41-
"float64]")
43+
"float64, uint8]")
4244

4345
self.data = []
4446
self.lod = []

0 commit comments

Comments
 (0)