From d8ee0d7b5c497f055eae53cc97a4d8dca9cfbd4a Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Thu, 3 Jun 2021 10:48:58 +0800 Subject: [PATCH] fix(mge/distributed): fix the mutli dataloader test error GitOrigin-RevId: 86c892591624beb210344600438ba534bb256954 --- imperative/python/megengine/data/dataloader.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/imperative/python/megengine/data/dataloader.py b/imperative/python/megengine/data/dataloader.py index 88b37c59..a9c226ea 100644 --- a/imperative/python/megengine/data/dataloader.py +++ b/imperative/python/megengine/data/dataloader.py @@ -45,7 +45,7 @@ def raise_timeout_error(): class DataLoader: r"""Provides a convenient way to iterate on a given dataset. - DataLoader combines a dataset with + DataLoader combines a dataset with :class:`~.Sampler`, :class:`~.Transform` and :class:`~.Collator`, make it flexible to get minibatch continually from a dataset. @@ -295,7 +295,7 @@ class _ParallelMapDataLoaderIter(_BaseMapDataLoaderIter): def _check_workers(self): # Check the status of each worker. if not self.data_collecting_worker.is_alive(): - exitcode = self.task_feeding_worker.exitcode + exitcode = self.data_collecting_worker.exitcode if exitcode != 0: raise RuntimeError("data collecting worker died. {}".format(exitcode))