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