fix progress and template

supervisely-ecosystem · Dec 13, 2024 · 3782da6 · 3782da6
1 parent 102ba38
commit 3782da6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/train/src/ui/monitoring.html b/train/src/ui/monitoring.html
@@ -112,7 +112,7 @@
 		</div>
 		<el-progress :percentage="data.progressPercentUploadDir"></el-progress>
 	</div>
-    <div v-if="state.benchmarkInProgress && data.progressBenchmark" class="mt10"></div>
+    <div v-if="state.benchmarkInProgress && data.progressBenchmark" class="mt10">
 		<div style="color: #20a0ff">
 			{{data.progressBenchmark}}: {{data.progressCurrentBenchmark}} /
 			{{data.progressTotalBenchmark}}

diff --git a/train/src/ui/monitoring.py b/train/src/ui/monitoring.py
@@ -478,14 +478,15 @@ def get_image_infos_by_split(split: list):
             train_images_ids = [img_info.id for img_info in train_image_infos]
 
         state["benchmarkInProgress"] = True
+        pbar = TqdmBenchmark
         bm = sly.nn.benchmark.SemanticSegmentationBenchmark(
             api,
             g.project_info.id,
             output_dir=g.data_dir + "/benchmark",
             gt_dataset_ids=benchmark_dataset_ids,
             gt_images_ids=benchmark_images_ids,
-            progress=TqdmBenchmark,
-            progress_secondary=TqdmBenchmark,
+            progress=pbar,
+            progress_secondary=pbar,
             classes_whitelist=classes,
         )