diff --git a/ppdet/data/reader.py b/ppdet/data/reader.py index f080f62c5635a664ae0aea753a1cbe80107728b9..144a84d452901df238d48c1b6f5d1036542d24da 100644 --- a/ppdet/data/reader.py +++ b/ppdet/data/reader.py @@ -180,7 +180,10 @@ class BaseDataLoader(object): else: self._batch_sampler = batch_sampler - use_shared_memory = self.use_shared_memory + # DataLoader do not start sub-process in Windows and Mac + # system, do not need to use shared memory + use_shared_memory = self.use_shared_memory and \ + sys.platform not in ['win32', 'darwin'] # check whether shared memory size is bigger than 1G(1024M) if use_shared_memory: shm_size = _get_shared_memory_size_in_M()