From 310cc10f71951d0b0b5feb5af213f3b646a19c38 Mon Sep 17 00:00:00 2001 From: liyuxiao2 Date: Thu, 22 May 2025 14:28:21 -0400 Subject: [PATCH] tweaked basic path --- data/training/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/data/training/train.py b/data/training/train.py index 92b74d2..ac2ca9c 100644 --- a/data/training/train.py +++ b/data/training/train.py @@ -36,10 +36,10 @@ def train_model(): checkpoint_filename = f"{training_cfg.checkpoint_path}" train_data = ( - f"{settings.dataset_cfg.output_dir}/train.h5" + f"{dataset_cfg.output_dir}/train.h5" ) val_data = ( - f"{settings.dataset_cfg.output_dir}/val.h5" + f"{dataset_cfg.output_dir}/val.h5" ) dataset_name = "Modulation Inference - Initial Model"