forked from qoherent/modrec-workflow
added the conf folder inside the repo
This commit is contained in:
parent
6c1164b466
commit
56eeee0354
26
conf/app.yaml
Normal file
26
conf/app.yaml
Normal 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"
|
|
@ -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(
|
||||
[
|
||||
|
|
Loading…
Reference in New Issue
Block a user