diff --git a/benchmark/run.py b/benchmark/run.py index f3c71de..6672eee 100644 --- a/benchmark/run.py +++ b/benchmark/run.py @@ -49,7 +49,7 @@ query_data = query_data[:query_size] if query_size else query_data # Warm up print('Warming up...') op = ops.image_embedding.timm(model_name=model_name, device=device).get_op() -dim = towhee.glob('https://raw.githubusercontent.com/towhee-io/towhee/main/towhee_logo.png') \ +dim = towhee.dc(['https://raw.githubusercontent.com/towhee-io/towhee/main/towhee_logo.png']) \ .image_decode() \ .image_embedding.timm(model_name=model_name, device=device)[0].shape[0] print(f'output dim: {dim}')