diff --git a/conf/app.yaml b/conf/app.yaml new file mode 100644 index 0000000..bd5f195 --- /dev/null +++ b/conf/app.yaml @@ -0,0 +1,26 @@ +general: + # Run mode. Options are 'prod' or 'dev'. + run_mode: prod + +dataset: + input_dir: "recordings" + num_slices: 8 + train_split: 0.8 + val_split : 0.2 + seed: 42 + output_dir: "data" + +training: + batch_size: 64 + epochs: 50 + learning_rate: 0.001 + checkpoint_path: "checkpoints/model.ckpt" + use_gpu: true + +inference: + model_path: "checkpoints/model.ckpt" + num_classes: 10 + results_output_path: "results/output.json" + +app: + build_dir: "dist" \ No newline at end of file diff --git a/scripts/produce_dataset.py b/scripts/produce_dataset.py index 4f7d233..7cce965 100644 --- a/scripts/produce_dataset.py +++ b/scripts/produce_dataset.py @@ -1,7 +1,7 @@ import os, h5py, numpy as np from utils.io import from_npy from split_dataset import split -from helpers.app_settings import get_app_settings +from ..helpers.app_settings import get_app_settings meta_dtype = np.dtype( [