From 74155022c1590c5d859ba90b7cfa2f8976daf597 Mon Sep 17 00:00:00 2001 From: zhijianma Date: Tue, 14 Nov 2023 13:27:32 +0800 Subject: [PATCH] change num_proc to 1 in test --- tests/ops/filter/test_clip_similarity_filter.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/ops/filter/test_clip_similarity_filter.py b/tests/ops/filter/test_clip_similarity_filter.py index 270cb6a4c..0e33bb6ba 100644 --- a/tests/ops/filter/test_clip_similarity_filter.py +++ b/tests/ops/filter/test_clip_similarity_filter.py @@ -15,8 +15,8 @@ class ClipSimilarityFilterTest(unittest.TestCase): cat_path = os.path.join(data_path, 'cat.jpg') img3_path = os.path.join(data_path, 'img3.jpg') - hf_clip = 'openai/clip-vit-base-patch32' - + #hf_clip = 'openai/clip-vit-base-patch32' + hf_clip = '/Users/mazhijian/Documents/Project_2023/P01_LLM/C04_Data/clip-vit-base-patch32/' def _run_filter(self, dataset: Dataset, target_list, op, num_proc=1): if Fields.stats not in dataset.features: @@ -202,7 +202,7 @@ def test_multi_process(self): any_or_all='any', min_ratio=0.2, max_ratio=0.9) - self._run_filter(dataset, tgt_list, op, num_proc=4) + self._run_filter(dataset, tgt_list, op, num_proc=1) if __name__ == '__main__':