fix dataset split

supervisely-ecosystem · Dec 16, 2024 · 146d484 · 146d484
1 parent 107050c
commit 146d484
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train/src/ui/monitoring.py b/train/src/ui/monitoring.py
@@ -507,6 +507,7 @@ def run_app():
                 val_datasets = state["valDatasets"]
                 benchmark_dataset_ids = [ds.id for ds in dataset_infos if ds.name in val_datasets]
                 train_dataset_ids = [ds.id for ds in dataset_infos if ds.name in train_datasets]
+                train_set, val_set = get_train_val_sets(g.project_dir, state)
             else:
 
                 def get_image_infos_by_split(split: list):