Skip to content

Commit

Permalink
update CPU benchmarks
Browse files Browse the repository at this point in the history
  • Loading branch information
Lednik7 authored May 3, 2022
1 parent fd61389 commit 9708705
Showing 1 changed file with 14 additions and 14 deletions.
28 changes: 14 additions & 14 deletions benchmark.md
Original file line number Diff line number Diff line change
@@ -1,21 +1,21 @@
# CPU benchmarks
Run on Intel (R) Xeon (R) CPU @ 2.30 GHz with 2 cores (Google Colab session)

| ONNX | batch | encode_image | encode_text | total |
|:------|--------:|---------------:|--------------:|----------:|
| ViT-B/32 | 2 | 0.399 | 0.219 | 0.618 |
| ViT-B/32 | 8 | 1.206 | 0.791 | 1.997 |
| ViT-B/32 | 16 | 2.342 | 1.542 | 3.884 |
| ViT-B/32 | 32 | 4.583 | 3.199 | 7.782 |
| ViT-B/32 | 64 | 9.372 | 6.885 | 16.257 |
| ONNX | batch | encode_image | encode_text | total |
|:---------|--------:|---------------:|--------------:|--------:|
| ViT-B/32 | 2 | 0.234 | 0.162 | 0.396 |
| ViT-B/32 | 8 | 0.923 | 0.656 | 1.579 |
| ViT-B/32 | 16 | 2.079 | 1.288 | 3.367 |
| ViT-B/32 | 32 | 3.937 | 2.658 | 6.595 |
| ViT-B/32 | 64 | 7.944 | 5.567 | 13.511 |

| TORCH | batch | encode_image | encode_text | total |
|:------|--------:|---------------:|--------------:|----------:|
| ViT-B/32 | 2 | 0.402 | 0.269 | 0.671 |
| ViT-B/32 | 8 | 1.21 | 0.849 | 2.059 |
| ViT-B/32 | 16 | 2.329 | 1.678 | 4.007 |
| ViT-B/32 | 32 | 4.662 | 3.492 | 8.154 |
| ViT-B/32 | 64 | 9.487 | 7.049 | 16.536 |
| TORCH | batch | encode_image | encode_text | total |
|:---------|--------:|---------------:|--------------:|--------:|
| ViT-B/32 | 2 | 0.343 | 0.243 | 0.586 |
| ViT-B/32 | 8 | 1.093 | 0.831 | 1.924 |
| ViT-B/32 | 16 | 1.952 | 1.523 | 3.475 |
| ViT-B/32 | 32 | 4.079 | 3.015 | 7.094 |
| ViT-B/32 | 64 | 8.07 | 6.212 | 14.282 |

# GPU benchmarks
Run on NVIDIA Tesla K80 (Google Colab session)
Expand Down

0 comments on commit 9708705

Please sign in to comment.