Update config.yaml

config.yaml

@@ -22,7 +22,7 @@ model:
   ff_dropout: 0
   flash_attn: True
   dim_freqs_in: 1025
-  sample_rate: 44100  # needed for mel filter bank from librosa
+  sample_rate: 44100
   stft_n_fft: 2048
   stft_hop_length: 441
   stft_win_length: 2048
@@ -45,26 +45,8 @@ training:
   instruments:
   - vocals
   - other
-  lr: 1.0
-  patience: 2
-  reduce_factor: 0.95
   target_instrument: vocals
-  num_epochs: 1000
-  num_steps: 1000
-  augmentation: false # enable augmentations by audiomentations and pedalboard
-  augmentation_type: true
-  use_mp3_compress: false # Deprecated
-  augmentation_mix: true # Mix several stems of the same type with some probability
-  augmentation_loudness: true # randomly change loudness of each stem
-  augmentation_loudness_type: 1 # Type 1 or 2
-  augmentation_loudness_min: 0.35
-  augmentation_loudness_max: 1
-  q: 0.99
-  coarse_loss_clip: false
-  ema_momentum: 0.999
-  optimizer: prodigy
-  other_fix: true # it's needed for checking on multisong dataset if other is actually instrumental
-  use_amp: true # enable or disable usage of mixed precision (float16) - usually it must be true
+  use_amp: true
 
 inference:
   batch_size: 4