add more support type

PaddlePaddle · Oct 15, 2020 · c1e3dbf · c1e3dbf · paddle-bot-old · Oct 15, 2020
1 parent 2714d76
commit c1e3dbf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/paddle/fluid/dataloader/dataloader_iter.py b/python/paddle/fluid/dataloader/dataloader_iter.py
@@ -17,6 +17,7 @@
 import sys
 import time
 import signal
+import numbers
 import logging
 import itertools
 import threading
@@ -83,7 +84,7 @@ def default_collate_fn(batch):
 
     outputs = []
     for slot in slots:
-        if isinstance(slot[0], np.ndarray):
+        if isinstance(slot[0], (np.ndarray, np.bool, numbers.Number)):
             tmp = np.stack(slot, axis=0)
             outputs.append(tmp)
         elif isinstance(slot[0], paddle.Tensor):