LiSenNet/config.yaml at main · hyyan2k/LiSenNet · GitHub

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
# gpu config
accelerator: gpu
devices: [0]

# log config
log_dir: ./log
resume: null  # if want to resume, specify ckpt path

# dataset config
dataset_config:
  batch_size: 4
  cut_len: 32000  # length of training samples: 2s
  num_workers: 4  # dataloader workers
  train_src_dir: <path-to-noisy-train-dir>
  train_tgt_dir: <path-to-clean-train-dir>
  val_src_dir: <path-to-noisy-val-dir>
  val_tgt_dir: <path-to-clean-val-dir>
  test_src_dir: <path-to-noisy-test-dir>
  test_tgt_dir: <path-to-clean-test-dir>


# training config
max_epochs: 100
val_check_interval: 0.5  # validate every 0.5 epochs
gradient_clip_val: 5.0
g_opt:
  lr: 5.0e-4
  betas: [0.8, 0.99]
d_opt:
  lr: 5.0e-4
  betas: [0.8, 0.99]
g_sch:
  step_size: 1
  gamma: 0.98
  verbose: true
d_sch:
  step_size: 1
  gamma: 0.98
  verbose: true

# model config and loss weights
model_config:
  num_channels: 16
  n_blocks: 2
  n_fft: 512
  hop_length: 256
  compress_factor: 0.3
weights:
  complex: 0.1
  mag: 0.9
  adv: 0.05