make training splits into its own pydantic base model, validate it sums to 1, make decoder script cleaner

2025-12-19 09:44:19 +01:00 · 2022-05-22 14:43:22 -07:00
parent 7ac2fc79f2
commit c6629c431a
3 changed files with 17 additions and 10 deletions
--- a/train_decoder.py
+++ b/train_decoder.py
@@ -422,9 +422,9 @@ def initialize_training(config):
    dataloaders = create_dataloaders (
        available_shards=all_shards,
        img_preproc = config.img_preproc,
-        train_prop = config.data["splits"]["train"],
-        val_prop = config.data["splits"]["val"],
-        test_prop = config.data["splits"]["test"],
+        train_prop = config.data.splits.train,
+        val_prop = config.data.splits.val,
+        test_prop = config.data.splits.test,
        n_sample_images=config.train.n_sample_images,
        **config.data.dict()
    )