2025-05-20 10:31:35 -04:00
|
|
|
general:
|
|
|
|
|
# Run mode. Options are 'prod' or 'dev'.
|
|
|
|
|
run_mode: prod
|
|
|
|
|
|
|
|
|
|
dataset:
|
2025-05-21 15:46:59 -04:00
|
|
|
input_dir: data/recordings
|
2025-05-20 10:31:35 -04:00
|
|
|
num_slices: 8
|
|
|
|
|
train_split: 0.8
|
|
|
|
|
val_split : 0.2
|
2025-05-21 15:46:59 -04:00
|
|
|
seed: 25
|
|
|
|
|
modulation_types: [bpsk, qpsk, qam16, qam64]
|
|
|
|
|
output_dir: data/dataset
|
2025-05-20 10:31:35 -04:00
|
|
|
|
|
|
|
|
training:
|
|
|
|
|
batch_size: 64
|
|
|
|
|
epochs: 50
|
|
|
|
|
learning_rate: 0.001
|
2025-05-21 15:46:59 -04:00
|
|
|
checkpoint_path: checkpoints/model.ckpt
|
2025-05-20 10:31:35 -04:00
|
|
|
use_gpu: true
|
|
|
|
|
|
|
|
|
|
inference:
|
2025-05-21 15:46:59 -04:00
|
|
|
model_path: checkpoints/model.ckpt
|
2025-05-20 10:31:35 -04:00
|
|
|
num_classes: 10
|
2025-05-21 15:46:59 -04:00
|
|
|
output_path: results/output.json
|
2025-05-20 10:31:35 -04:00
|
|
|
|
|
|
|
|
app:
|
2025-05-21 15:46:59 -04:00
|
|
|
build_dir: dist
|
