-
Notifications
You must be signed in to change notification settings - Fork 4
Expand file tree
/
Copy pathconfig.yaml
More file actions
170 lines (163 loc) · 3.46 KB
/
config.yaml
File metadata and controls
170 lines (163 loc) · 3.46 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
epochs: 2000
data_dir: null
log_dir: null
lufs: -18
sr: 44100
chunk_duration: 12
chunk_overlap: 5
device: cuda
batch_size: 35
dataset: medley_vocal
regularise_delay: true
enable_progress_bar: true
model:
_target_: torch.nn.Sequential
_args_:
- _target_: modules.fx.Peak
sr: ${sr}
freq: 800
min_freq: 33
max_freq: 5400
- _target_: modules.fx.Peak
sr: ${sr}
freq: 4000
min_freq: 200
max_freq: 17500
- _target_: modules.fx.LowShelf
sr: ${sr}
freq: 115
min_freq: 30
max_freq: 200
- _target_: modules.fx.HighShelf
sr: ${sr}
freq: 6000
min_freq: 750
max_freq: 8300
- _target_: modules.fx.LowPass
sr: ${sr}
freq: 17500
min_freq: 200
max_freq: 18000
- _target_: modules.fx.HighPass
sr: ${sr}
freq: 200
min_freq: 16
max_freq: 5300
- _target_: modules.fx.CompressorExpander
sr: ${sr}
cmp_ratio: 2.0
exp_ratio: 0.5
at_ms: 50.0
rt_ms: 50.0
avg_coef: 0.3
cmp_th: -18.0
exp_th: -48.0
make_up: 0.0
lookahead: true
max_lookahead: 15
- _target_: modules.fx.SendFXsAndSum
_args_:
- _target_: modules.fx.SurrogateDelay
sr: ${sr}
delay: 400
dropout: 0
straight_through: true
recursive_eq: true
ir_duration: 4
eq:
_target_: modules.fx.LowPass
sr: ${sr}
freq: 8000
min_freq: 200
max_freq: 16000
min_Q: 0.5
max_Q: 2
- _target_: modules.fx.FDN
sr: ${sr}
delays:
- 997
- 1153
- 1327
- 1559
- 1801
- 2099
num_decay_freq: 49
delay_independent_decay: true
ir_duration: 12
eq:
_target_: torch.nn.Sequential
_args_:
- _target_: modules.fx.Peak
sr: ${sr}
freq: 800
min_freq: 200
max_freq: 2500
min_Q: 0.1
max_Q: 3
- _target_: modules.fx.Peak
sr: ${sr}
freq: 4000
min_freq: 600
max_freq: 7000
min_Q: 0.1
max_Q: 3
- _target_: modules.fx.LowShelf
sr: ${sr}
freq: 115
min_freq: 30
max_freq: 450
- _target_: modules.fx.HighShelf
sr: ${sr}
freq: 8000
min_freq: 1500
max_freq: 16000
cross_send: true
pan_direct: true
optimiser:
_target_: torch.optim.Adam
lr: 0.01
mss:
fft_sizes:
- 128
- 512
- 2048
hop_sizes:
- 32
- 128
- 512
mldr:
s_taus:
- 50
- 100
l_taus:
- 1000
- 2000
loss_fn:
_target_: loss.SumLosses
weights:
- 1.0
- 0.5
- 0.5
- 0.25
loss_fns:
- _target_: auraloss.freq.MultiResolutionSTFTLoss
fft_sizes: ${mss.fft_sizes}
hop_sizes: ${mss.hop_sizes}
win_lengths: ${.fft_sizes}
sample_rate: ${sr}
perceptual_weighting: true
- _target_: auraloss.freq.SumAndDifferenceSTFTLoss
fft_sizes: ${mss.fft_sizes}
hop_sizes: ${mss.hop_sizes}
win_lengths: ${.fft_sizes}
sample_rate: ${sr}
perceptual_weighting: true
- _target_: loss.ldr.MLDRLoss
sr: ${sr}
s_taus: ${mldr.s_taus}
l_taus: ${mldr.l_taus}
- _target_: loss.ldr.MLDRLoss
sr: ${sr}
mid_side: true
s_taus: ${mldr.s_taus}
l_taus: ${mldr.l_taus}