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

[Fix] Disable worker in the dataloader of gpu unit test

Browse Source
This commit is contained in:
Tong Gao 2022-02-16 20:42:22 +08:00 committed by GitHub
parent fc9d73761b
commit 0f5c7d38f8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_apis/test_single_gpu_test.py
View File

@ -43,7 +43,7 @@ def generate_sample_dataloader(cfg, curr_dir, img_prefix='', ann_file=''):
test = copy.deepcopy(cfg.data.test.datasets[0]) test = copy.deepcopy(cfg.data.test.datasets[0])
test.img_prefix = img_prefix test.img_prefix = img_prefix
test.ann_file = ann_file test.ann_file = ann_file
cfg.data.workers_per_gpu = 1 cfg.data.workers_per_gpu = 0
cfg.data.test.datasets = [test] cfg.data.test.datasets = [test]
dataset = build_dataset(cfg.data.test) dataset = build_dataset(cfg.data.test)