update resource limit (open-mmlab#783)

wjkim81 · Jul 19, 2021 · fe4fdad · fe4fdad
1 parent f1ec111
commit fe4fdad
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mmpose/datasets/builder.py b/mmpose/datasets/builder.py
@@ -14,8 +14,9 @@
     # 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')