From 20c84617957b6efb62a5c05d3f75ceb065b70765 Mon Sep 17 00:00:00 2001 From: "min.tian" Date: Thu, 23 Jan 2025 15:30:34 +0800 Subject: [PATCH] Metrics will separately record the duration of insert and optimize. Signed-off-by: min.tian --- vectordb_bench/backend/task_runner.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/vectordb_bench/backend/task_runner.py b/vectordb_bench/backend/task_runner.py index 90ccf762..a50f118c 100644 --- a/vectordb_bench/backend/task_runner.py +++ b/vectordb_bench/backend/task_runner.py @@ -154,6 +154,8 @@ def _run_perf_case(self, drop_old: bool = True) -> Metric: if TaskStage.LOAD in self.config.stages: _, load_dur = self._load_train_data() build_dur = self._optimize() + m.insert_duration = round(load_dur, 4) + m.optimize_duration = round(build_dur, 4) m.load_duration = round(load_dur + build_dur, 4) log.info( f"Finish loading the entire dataset into VectorDB,"