Revert "fix dataloaer for toolkit (PaddlePaddle#61867) (PaddlePaddle#…

…61994)" This reverts commit b50e906.
hanhaowen-mt · May 13, 2024 · e627c5b · e627c5b
1 parent 126876b
commit e627c5b
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/python/paddle/io/dataloader/dataloader_iter.py b/python/paddle/io/dataloader/dataloader_iter.py
@@ -704,11 +704,10 @@ def _get_data(self):
                 if len(failed_workers) > 0:
                     self._exit_thread_unexpectedly()
                     pids = ', '.join(str(w.pid) for w in failed_workers)
-                    logging.warning(
-                        "DataLoader {} workers exit unexpectedly, "
-                        "pids: {}".format(len(failed_workers), pids)
+                    raise RuntimeError(
+                        f"DataLoader {len(failed_workers)} workers exit unexpectedly, "
+                        f"pids: {pids}"
                     )
-                    return
 
                 # get(timeout) will call _poll(timeout) and may raise IOError
                 if isinstance(e, (IOError, queue.Empty)):