E009 Sheffield
Python Modules
Configuration Files
config.yaml
1path:
2 cec1_root: ???
3 exp_folder: ??? # folder to store intermediate signals and final results
4
5sample_rate: 44100
6downsample_factor: 2
7num_channels: 6
8level_norm: False
9wav_silence_len: 2
10
11listener:
12 metafile: ${path.cec1_root}/clarity_CEC1_data/clarity_data/metadata/listeners.json
13 id: L0001
14 src_pos: ff
15 loudnorm: False
16
17mc_conv_tasnet:
18 N_spec: 256
19 N_spat: 128 # 6 * 30
20 L: 20
21 B: 256
22 H: 512
23 P: 3
24 X: 6
25 R: 4
26 C: 1 # num_speakers
27 num_channels: 6 # should be consistent with dataloader num_channels
28 norm_type: cLN
29 causal: True
30 mask_nonlinear: relu
31
32fir:
33 sample_rate: ${sample_rate}
34 nfir: 220
35
36den_trainer:
37 optimizer:
38 lr: 1e-3
39 epochs: 200
40 gradient_clip_val: 5.0
41
42amp_trainer:
43 optimizer:
44 lr: 5e-2
45 epochs: 100
46 gradient_clip_val: 5.0
47 stoilevel_loss:
48 sample_rate: ${sample_rate}
49 alpha: 1e-4
50 num_sanity_val_steps: 2
51
52train_dataset:
53 scenes_folder: ${path.cec1_root}/clarity_CEC1_data/clarity_data/train/scenes
54 scenes_file: ${path.cec1_root}/clarity_CEC1_data/clarity_data/metadata/scenes.train.json
55 sample_rate: ${sample_rate}
56 downsample_factor: ${downsample_factor}
57 wav_sample_len: 2
58 wav_silence_len: ${wav_silence_len}
59 num_channels: ${num_channels}
60 norm: ${level_norm}
61
62train_loader:
63 batch_size: 2
64 shuffle: True
65 num_workers: 10
66 drop_last: true
67 pin_memory: True
68
69dev_dataset:
70 scenes_folder: ${path.cec1_root}/clarity_CEC1_data/clarity_data/dev/scenes
71 scenes_file: ${path.cec1_root}/clarity_CEC1_data/clarity_data/metadata/scenes.dev.json
72 sample_rate: ${sample_rate}
73 downsample_factor: ${downsample_factor}
74 wav_sample_len: null
75 wav_silence_len: ${wav_silence_len}
76 num_channels: ${num_channels}
77 norm: ${level_norm}
78
79dev_loader:
80 batch_size: 1
81 shuffle: False
82 num_workers: 10
83
84test_dataset:
85 scenes_folder: ${path.cec1_root}/clarity_CEC1_data_eval/clarity_data/eval/scenes
86 scenes_file: ${path.cec1_root}/clarity_CEC1_data_eval/clarity_data/metadata/scenes_listeners.eval.json
87 sample_rate: ${sample_rate}
88 downsample_factor: ${downsample_factor}
89 wav_sample_len: null
90 wav_silence_len: 0
91 num_channels: ${num_channels}
92 norm: ${level_norm}
93 testing: True
94
95test_loader:
96 batch_size: 1
97 shuffle: False
98 num_workers: 10
99
100hydra:
101 # output_subdir: ${path.exp_folder}.hydra
102 run:
103 dir: ${path.exp_folder}
104 job:
105 chdir: True