From dc296f64c65dfbb314bf3ae456e747e3b171cdb4 Mon Sep 17 00:00:00 2001 From: LXXXXR <73265258+LXXXXR@users.noreply.github.com> Date: Sun, 16 May 2021 21:53:13 +0800 Subject: [PATCH] [Fix] Fix multi-node test tmp dir (#251) * fix multi-node test tmp dir * fix mmcv version --- mmcls/apis/test.py | 3 ++- requirements/mminstall.txt | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/mmcls/apis/test.py b/mmcls/apis/test.py index ec789b9ef..a7df59729 100644 --- a/mmcls/apis/test.py +++ b/mmcls/apis/test.py @@ -126,7 +126,8 @@ def collect_results_cpu(result_part, size, tmpdir=None): dtype=torch.uint8, device='cuda') if rank == 0: - tmpdir = tempfile.mkdtemp() + mmcv.mkdir_or_exist('.dist_test') + tmpdir = tempfile.mkdtemp(dir='.dist_test') tmpdir = torch.tensor( bytearray(tmpdir.encode()), dtype=torch.uint8, device='cuda') dir_tensor[:len(tmpdir)] = tmpdir diff --git a/requirements/mminstall.txt b/requirements/mminstall.txt index d371e1cc8..521afb155 100644 --- a/requirements/mminstall.txt +++ b/requirements/mminstall.txt @@ -1 +1 @@ -mmcv-full>=1.3.0 +mmcv-full>=1.3.0,<1.5.0