From 00defd6b7fa673cd2357bb17e12c6d3d07cbd507 Mon Sep 17 00:00:00 2001 From: Junjun2016 Date: Thu, 15 Jul 2021 11:43:21 +0800 Subject: [PATCH] update resource limit (#700) --- mmseg/datasets/builder.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mmseg/datasets/builder.py b/mmseg/datasets/builder.py index e6284e5c9..5994ab233 100644 --- a/mmseg/datasets/builder.py +++ b/mmseg/datasets/builder.py @@ -14,8 +14,9 @@ if platform.system() != 'Windows': # https://github.com/pytorch/pytorch/issues/973 import resource rlimit = resource.getrlimit(resource.RLIMIT_NOFILE) + base_soft_limit = rlimit[0] hard_limit = rlimit[1] - soft_limit = min(4096, hard_limit) + soft_limit = min(max(4096, base_soft_limit), hard_limit) resource.setrlimit(resource.RLIMIT_NOFILE, (soft_limit, hard_limit)) DATASETS = Registry('dataset')