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

[Fix] Fix multi-node test tmp dir (#251)

Browse Source 
* fix multi-node test tmp dir

* fix mmcv version
This commit is contained in:
LXXXXR 2021-05-16 21:53:13 +08:00 committed by GitHub
parent 5ee08767f2
commit dc296f64c6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mmcls/apis/test.py
View File

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

2
requirements/mminstall.txt
View File

@ -1 +1 @@
mmcv-full>=1.3.0
mmcv-full>=1.3.0,<1.5.0