added the conf folder inside the repo

This commit is contained in:
liyuxiao2 2025-05-20 10:31:35 -04:00
parent 6c1164b466
commit 56eeee0354
2 changed files with 27 additions and 1 deletions

26
conf/app.yaml Normal file
View File

@ -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"

View File

@ -1,7 +1,7 @@
import os, h5py, numpy as np import os, h5py, numpy as np
from utils.io import from_npy from utils.io import from_npy
from split_dataset import split 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( meta_dtype = np.dtype(
[ [