aboutsummaryrefslogtreecommitdiff
path: root/deeptagger
diff options
context:
space:
mode:
authorPřemysl Eric Janouch <p@janouch.name>2024-01-19 15:32:10 +0100
committerPřemysl Eric Janouch <p@janouch.name>2024-01-19 15:33:49 +0100
commitfd5e3bb166638a9666ce632d5965f46745d320a9 (patch)
tree770a4e58a3f3a0990ea92df5e5272345f088b9aa /deeptagger
parent77e988365d1c3a5ee67301c99c8992fbed7ce7bd (diff)
downloadgallery-fd5e3bb166638a9666ce632d5965f46745d320a9.tar.gz
gallery-fd5e3bb166638a9666ce632d5965f46745d320a9.tar.xz
gallery-fd5e3bb166638a9666ce632d5965f46745d320a9.zip
Add CoreML benchmarks
Diffstat (limited to 'deeptagger')
-rw-r--r--deeptagger/README.adoc90
1 files changed, 86 insertions, 4 deletions
diff --git a/deeptagger/README.adoc b/deeptagger/README.adoc
index 8d65dfe..9a4d3aa 100644
--- a/deeptagger/README.adoc
+++ b/deeptagger/README.adoc
@@ -49,11 +49,11 @@ Options
--threshold 0.1::
Output weight threshold. Needs to be set very high on ML-Danbooru models.
-Model benchmarks
-----------------
-These were measured on a machine with GeForce RTX 4090 (24G),
+Model benchmarks (Linux)
+------------------------
+These were measured with ORT 1.16.3 on a machine with GeForce RTX 4090 (24G),
and Ryzen 9 7950X3D (32 threads), on a sample of 704 images,
-which took over eight hours.
+which took over eight hours. Times include model loading.
There is room for further performance tuning.
@@ -128,3 +128,85 @@ CPU inference
|ML-Danbooru Caformer dec-5-97527|16|689 s
|ML-Danbooru Caformer dec-5-97527|1|829 s
|===
+
+Model benchmarks (macOS)
+------------------------
+These were measured with ORT 1.16.3 on a MacBook Pro, M1 Pro (16GB),
+macOS Ventura 13.6.2, on a sample of 179 images. Times include model loading.
+
+There was often significant memory pressure and swapping,
+which may explain some of the anomalies. CoreML often makes things worse,
+and generally consumes a lot more memory than pure CPU execution.
+
+The kernel panic was repeatable.
+
+GPU inference
+~~~~~~~~~~~~~
+[cols="<,>,>", options=header]
+|===
+|Model|Batch size|Time
+|DeepDanbooru|1|24 s
+|DeepDanbooru|8|31 s
+|DeepDanbooru|4|33 s
+|WD v1.4 SwinV2 v2 (batch)|4|71 s
+|WD v1.4 SwinV2 v2 (batch)|1|76 s
+|WD v1.4 ViT v2 (batch)|4|97 s
+|WD v1.4 ViT v2 (batch)|8|97 s
+|ML-Danbooru TResNet-D 6-30000|8|100 s
+|ML-Danbooru TResNet-D 6-30000|4|101 s
+|WD v1.4 ViT v2 (batch)|1|105 s
+|ML-Danbooru TResNet-D 6-30000|1|125 s
+|WD v1.4 ConvNeXT v2 (batch)|8|126 s
+|WD v1.4 SwinV2 v2 (batch)|8|127 s
+|WD v1.4 ConvNeXT v2 (batch)|4|128 s
+|WD v1.4 ConvNeXTV2 v2 (batch)|8|132 s
+|WD v1.4 ConvNeXTV2 v2 (batch)|4|133 s
+|WD v1.4 ViT v2|1|146 s
+|WD v1.4 ConvNeXT v2 (batch)|1|149 s
+|WD v1.4 ConvNeXTV2 v2 (batch)|1|160 s
+|WD v1.4 MOAT v2 (batch)|1|165 s
+|WD v1.4 SwinV2 v2|1|166 s
+|WD v1.4 ConvNeXT v2|1|273 s
+|WD v1.4 MOAT v2|1|273 s
+|WD v1.4 ConvNeXTV2 v2|1|340 s
+|ML-Danbooru Caformer dec-5-97527|1|551 s
+|ML-Danbooru Caformer dec-5-97527|4|swap hell
+|ML-Danbooru Caformer dec-5-97527|8|swap hell
+|WD v1.4 MOAT v2 (batch)|4|kernel panic
+|===
+
+CPU inference
+~~~~~~~~~~~~~
+[cols="<,>,>", options=header]
+|===
+|Model|Batch size|Time
+|DeepDanbooru|8|54 s
+|DeepDanbooru|4|55 s
+|DeepDanbooru|1|75 s
+|WD v1.4 SwinV2 v2 (batch)|8|93 s
+|WD v1.4 SwinV2 v2 (batch)|4|94 s
+|ML-Danbooru TResNet-D 6-30000|8|97 s
+|WD v1.4 SwinV2 v2 (batch)|1|98 s
+|ML-Danbooru TResNet-D 6-30000|4|99 s
+|WD v1.4 SwinV2 v2|1|99 s
+|WD v1.4 ViT v2 (batch)|4|111 s
+|WD v1.4 ViT v2 (batch)|8|111 s
+|WD v1.4 ViT v2 (batch)|1|113 s
+|WD v1.4 ViT v2|1|113 s
+|ML-Danbooru TResNet-D 6-30000|1|118 s
+|WD v1.4 ConvNeXT v2 (batch)|8|124 s
+|WD v1.4 ConvNeXT v2 (batch)|4|125 s
+|WD v1.4 ConvNeXTV2 v2 (batch)|8|129 s
+|WD v1.4 ConvNeXT v2|1|130 s
+|WD v1.4 ConvNeXTV2 v2 (batch)|4|131 s
+|WD v1.4 MOAT v2 (batch)|8|134 s
+|WD v1.4 ConvNeXTV2 v2|1|136 s
+|WD v1.4 MOAT v2 (batch)|4|136 s
+|WD v1.4 ConvNeXT v2 (batch)|1|146 s
+|WD v1.4 MOAT v2 (batch)|1|156 s
+|WD v1.4 MOAT v2|1|156 s
+|WD v1.4 ConvNeXTV2 v2 (batch)|1|157 s
+|ML-Danbooru Caformer dec-5-97527|4|241 s
+|ML-Danbooru Caformer dec-5-97527|8|241 s
+|ML-Danbooru Caformer dec-5-97527|1|262 s
+|===