logo

image-embedding
/
timm
repo-copy-icon

copied

Browse Source

Update due to change of triton_client 

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
Jael Gu 2 years ago
parent
158d86d00e
commit
4f2ad553b6
1 changed files with 1 additions and 1 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 2
      benchmark/qps_test.py

2
benchmark/qps_test.py
View File

@ -66,7 +66,7 @@ if args.num and args.pipe:
if args.triton: if args.triton:
client = triton_client.Client(url='localhost:8000') client = triton_client.Client(url='localhost:8000')
out2 = client(data)[0][0][0]
out2 = client(data)[0][0]
print('Triton: OK') print('Triton: OK')
if numpy.allclose(out1, out2, atol=args.atol): if numpy.allclose(out1, out2, atol=args.atol):

Write Preview
Loading…
Cancel
Save