michael/modrec-workflow
1
0
Fork 0
forked from qoherent/modrec-workflow
Code Pull Requests Workflows Activity

tweaked basic path

Browse Source
This commit is contained in:
liyuxiao2 2025-05-22 14:28:21 -04:00
parent 5970af5be7
commit 310cc10f71
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
data/training/train.py
View File

@ -36,10 +36,10 @@ def train_model():
checkpoint_filename = f"{training_cfg.checkpoint_path}" checkpoint_filename = f"{training_cfg.checkpoint_path}"
train_data = ( train_data = (
f"{settings.dataset_cfg.output_dir}/train.h5" f"{dataset_cfg.output_dir}/train.h5"
) )
val_data = ( val_data = (
f"{settings.dataset_cfg.output_dir}/val.h5" f"{dataset_cfg.output_dir}/val.h5"
) )
dataset_name = "Modulation Inference - Initial Model" dataset_name = "Modulation Inference - Initial Model"