final

Author: PINTO0309

benchmark/benchmark_times_combined_2.png

@@ -0,0 +1,11 @@
+run,avg_ms
+1,0.56
+2,0.63
+3,0.69
+4,0.81
+5,0.87
+6,0.98
+7,1.03
+8,1.19
+9,1.24
+10,1.32

benchmark/benchmark_times_gazelle_hgnetv2_atto_distill_1x3x320x320_1xNx4.csv

@@ -0,0 +1,11 @@
+run,avg_ms
+1,0.72
+2,0.87
+3,1.03
+4,1.15
+5,1.35
+6,1.48
+7,1.63
+8,1.79
+9,2.03
+10,2.10

benchmark/benchmark_times_gazelle_hgnetv2_atto_distill_1x3x320x320_1xNx4.png

@@ -0,0 +1,11 @@
+run,avg_ms
+1,1.62
+2,2.02
+3,2.45
+4,2.96
+5,3.33
+6,3.79
+7,4.17
+8,4.67
+9,5.03
+10,5.50

benchmark/benchmark_times_gazelle_hgnetv2_femto_distill_1x3x416x416_1xNx4.csv

@@ -0,0 +1,11 @@
+run,avg_ms
+1,1.65
+2,2.05
+3,2.41
+4,2.91
+5,3.33
+6,3.78
+7,4.20
+8,4.69
+9,5.15
+10,5.45

benchmark/benchmark_times_gazelle_hgnetv2_femto_distill_1x3x416x416_1xNx4.png

@@ -42,10 +42,10 @@ def get_transform(self, in_size):
     "gazelle_hgnetv2_pico": ["ckpts/gazelle_hgnetv2_pico.pt", False, 640, 640],
     "gazelle_hgnetv2_n": ["ckpts/gazelle_hgnetv2_n.pt", False, 640, 640],
 
-    "gazelle_hgnetv2_atto_inout": ["ckpts/gazelle_hgnetv2_atto_inout.pt", True, 320, 320],
-    "gazelle_hgnetv2_femto_inout": ["ckpts/gazelle_hgnetv2_femto_inout.pt", True, 416, 416],
-    "gazelle_hgnetv2_pico_inout": ["ckpts/gazelle_hgnetv2_pico_inout.pt", True, 640, 640],
-    "gazelle_hgnetv2_n_inout": ["ckpts/gazelle_hgnetv2_n_inout.pt", True, 640, 640],
+    "gazelle_hgnetv2_atto_inout": ["ckpts/gazelle_hgnetv2_atto_inout_distill.pt", True, 320, 320],
+    "gazelle_hgnetv2_femto_inout": ["ckpts/gazelle_hgnetv2_femto_inout_distill.pt", True, 416, 416],
+    "gazelle_hgnetv2_pico_inout": ["ckpts/gazelle_hgnetv2_pico_inout_distill.pt", True, 640, 640],
+    "gazelle_hgnetv2_n_inout": ["ckpts/gazelle_hgnetv2_n_inout_distill.pt", True, 640, 640],
 
     "gazelle_dinov3_vit_tiny": ["ckpts/gazelle_dinov3_vit_tiny.pt", False, 640, 640],
     "gazelle_dinov3_vit_tinyplus": ["ckpts/gazelle_dinov3_vit_tinyplus.pt", False, 640, 640],