Spaces:
Running
Running
| 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 | |