toolbox/torchaudio/models/lstm/yaml/config.yaml

model_name: "lstm"

# spec
sample_rate: 8000
segment_size: 32000
n_fft: 320
win_size: 320
hop_size: 160
win_type: hann

# data
max_snr_db: 20
min_snr_db: -10

# model
hidden_size: 512
num_layers: 3
dropout: 0.1

# train
max_epochs: 100
batch_size: 32
num_workers: 4
seed: 1234

lr: 0.001
lr_scheduler: CosineAnnealingLR
lr_scheduler_kwargs: {}

weight_decay: 0.00001
clip_grad_norm: 10.0
eval_steps: 25000