| --- |
| task_name: AudioSep |
|
|
| data: |
| datafiles: |
| - 'datafiles/template.json' |
|
|
| sampling_rate: 32000 |
| segment_seconds: 5 |
| loudness_norm: |
| lower_db: -10 |
| higher_db: 10 |
| max_mix_num: 2 |
|
|
| model: |
| query_net: CLAP |
| condition_size: 512 |
| model_type: ResUNet30 |
| input_channels: 1 |
| output_channels: 1 |
| resume_checkpoint: "" |
| use_text_ratio: 1.0 |
|
|
| train: |
| optimizer: |
| optimizer_type: AdamW |
| learning_rate: 1e-3 |
| warm_up_steps: 10000 |
| reduce_lr_steps: 1000000 |
| lr_lambda_type: constant_warm_up |
| num_nodes: 1 |
| num_workers: 6 |
| loss_type: l1_wav |
| sync_batchnorm: True |
| batch_size_per_device: 12 |
| steps_per_epoch: 10000 |
| evaluate_step_frequency: 10000 |
| save_step_frequency: 20000 |
| early_stop_steps: 10000001 |
| random_seed: 1234 |
|
|
|
|
