| audio: | |
| chunk_size: 132300 | |
| num_channels: 2 | |
| sample_rate: 44100 | |
| min_mean_abs: 0.0 | |
| model: | |
| sr: 44100 | |
| win: 20 | |
| feature_dim: 384 | |
| layer: 6 | |
| training: | |
| instruments: ['restored', 'addition'] | |
| target_instrument: 'restored' | |
| batch_size: 2 | |
| num_steps: 1000 | |
| num_epochs: 1000 | |
| optimizer: 'prodigy' | |
| lr: 1.0 | |
| patience: 2 | |
| reduce_factor: 0.95 | |
| coarse_loss_clip: true | |
| grad_clip: 0 | |
| q: 0.95 | |
| use_amp: true | |
| augmentations: | |
| enable: false # enable or disable all augmentations (to fast disable if needed) | |
| inference: | |
| batch_size: 4 | |
| num_overlap: 4 |