added the conf folder inside the repo
Some checks failed
RIA Hub Workflow Demo / ria-demo (push) Failing after 15s
Some checks failed
RIA Hub Workflow Demo / ria-demo (push) Failing after 15s
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
|
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(
|
||||||
[
|
[
|
||||||
|
|
Loading…
Reference in New Issue
Block a user