SESAME/config.yaml at main · nullpath-ml/SESAME · 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
52
53
54
55
56
57
58
59
60
61
model:
  dense_channel: 64
  num_tsblocks: 4
  n_heads: 4
  dense_depth: 4
  beta: 2.0
  compress_factor: 0.3
  disc_dim: 16
  disc_dropout: 0.3
  moe:
    apply_to: [0, 1]
    num_experts: 4
    top_k: 2
    expert_ffn_dim: 256
    balance_loss_weight: 0.01
    z_loss_weight: 0.001
    bias_update_speed: 0.001
    noise_ctx_dim: 64

training:
  learning_rate: 0.0005
  adam_b1: 0.8
  adam_b2: 0.99
  batch_size: 2
  epochs: 400
  warmup_steps: 58000
  loss_weights:
    mag: 0.9
    pha: 0.3
    com: 0.1
    stft: 0.1
    metric: 0.05
    time: 0.2
  train_pesq_workers: 16

intervals:
  stdout_interval: 5
  summary_interval: 50
  checkpoint_interval: 5000
  validation_interval: 5000
  val_pesq_workers: 30
  best_checkpoint_start_epoch: 16

data:
  max_validation_samples: 7000
  sampling_rate: 16000
  segment_size: 32000
  n_fft: 400
  hop_size: 100
  win_size: 400
  num_workers: 4

paths:
  checkpoint_path: cp_model
  input_clean_wavs_dir: /work/data/VoiceBank+DEMAND/wav_clean
  input_noisy_wavs_dir: /work/data/VoiceBank+DEMAND/wav_noisy
  input_training_file: /work/data/VoiceBank+DEMAND/training.txt
  input_validation_file: /work/data/VoiceBank+DEMAND/test.txt

reproducibility:
  seed: 1234