From 16ef54c45dfd3e553d6c11d66b7a2bc8d66e4db0 Mon Sep 17 00:00:00 2001 From: RangiLyu Date: Thu, 21 Jul 2022 16:02:51 +0800 Subject: [PATCH] [Fix] Fix compatibility of persistent_workers. (#379) --- mmengine/runner/runner.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/mmengine/runner/runner.py b/mmengine/runner/runner.py index 4ae1c086..a38ee678 100644 --- a/mmengine/runner/runner.py +++ b/mmengine/runner/runner.py @@ -1382,6 +1382,13 @@ class Runner: else: init_fn = None + # `persistent_workers` requires pytorch version >= 1.7 + if ('persistent_workers' in dataloader_cfg + and digit_version(TORCH_VERSION) < digit_version('1.7.0')): + warnings.warn('`persistent_workers` is only available when ' + 'pytorch version >= 1.7') + dataloader_cfg.pop('persistent_workers') + # The default behavior of `collat_fn` in dataloader is to # merge a list of samples to form a mini-batch of Tensor(s). # However, to make this more flexible, collate_fn in MMengine does