modrec-workflow/conf/app.yaml

40 lines
836 B
YAML
Raw Normal View History

2025-05-20 10:31:35 -04:00
general:
# Run mode. Options are 'prod' or 'dev'.
run_mode: prod
dataset:
2025-06-13 13:58:35 -04:00
#number of slices you want to split each recording into
2025-05-20 10:31:35 -04:00
num_slices: 8
2025-06-13 13:58:35 -04:00
#training/val split between the 2 data sets
2025-05-20 10:31:35 -04:00
train_split: 0.8
val_split : 0.2
2025-06-13 13:58:35 -04:00
#used to initialize a random number generator.
2025-05-21 15:46:59 -04:00
seed: 25
2025-06-13 13:58:35 -04:00
#multiple modulations to contain in the dataset
2025-05-21 15:46:59 -04:00
modulation_types: [bpsk, qpsk, qam16, qam64]
2025-06-13 13:58:35 -04:00
2025-05-20 10:31:35 -04:00
training:
2025-06-13 13:58:35 -04:00
#number of training samples being processed together before model updates its weights
batch_size: 256
2025-06-13 13:58:35 -04:00
#number of passes through the data set during the training process
epochs: 5
2025-06-13 13:58:35 -04:00
#how much the weights update during training after every batch
#suggested range for fine-tuning: (1e-6, 1e-4)
learning_rate: 1e-4
2025-06-13 13:58:35 -04:00
2025-05-20 10:31:35 -04:00
use_gpu: true
inference:
2025-06-13 13:58:35 -04:00
#num classes to classify on
num_classes: 4
2025-06-13 13:58:35 -04:00
2025-05-20 10:31:35 -04:00
2025-06-13 14:17:13 -04:00
2025-05-20 10:31:35 -04:00
app:
2025-05-21 15:46:59 -04:00
build_dir: dist