forked from qoherent/modrec-workflow
26 lines
453 B
YAML
26 lines
453 B
YAML
|
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"
|