From 146d484a76a19a5aa957e70b1b07444d240c9b90 Mon Sep 17 00:00:00 2001 From: almaz Date: Mon, 16 Dec 2024 12:13:16 +0100 Subject: [PATCH] fix dataset split --- train/src/ui/monitoring.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train/src/ui/monitoring.py b/train/src/ui/monitoring.py index b61bd5a..0afbb98 100644 --- 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):