Local
Python Modules
Configuration Files
conf.yml
1# ConvTasNet config
2convtasnet:
3 N: 256
4 L: 20
5 B: 256
6 H: 512
7 P: 3
8 X: 10
9 R: 4
10 C: 2
11 audio_channels: 2 # stereo
12 norm_type: gLN
13 causal: False
14 mask_nonlinear: 'relu'
15
16# Training config
17training:
18 epochs: 200
19 batch_size: 4 # Aggregate 2 iteration, effective batch size = 8
20 num_workers: 4
21 half_lr: yes
22 early_stop: yes
23 aggregate: 2 # aggregate gradiante every 2 iterations
24# Optim config
25optim:
26 lr: 0.0005
27# Data config
28data:
29 root: /path/to/MUSDB18
30 mix_background: true
31 segment: 5.0 # Seconds
32 samples_per_track: 64
33 sample_rate: 44100