mirrors/mmengine
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmengine.git synced 2025-06-03 21:54:44 +08:00
Code Issues Projects Releases Wiki Activity

[Fix] Fix compatibility of persistent_workers. (#379)

Browse Source
This commit is contained in:
RangiLyu 2022-07-21 16:02:51 +08:00 committed by GitHub
parent 6bd548d8bb
commit 16ef54c45d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mmengine/runner/runner.py
View File

@ -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