Update README.md

README.md

@@ -13,8 +13,14 @@ Try it with [ZFTurbo's Music-Source-Separation-Training](https://github.com/ZFTu
 
 ### Model
 
-Configs: [config_dereverb-echo_mel_band_roformer.yaml](./config_dereverb-echo_mel_band_roformer.yaml)<br>
-Model: [dereverb-echo_mel_band_roformer_sdr_10.0169.ckpt](./dereverb-echo_mel_band_roformer_sdr_10.0169.ckpt)<br>
+Configs_256_8_4: [config_dereverb-echo_mel_band_roformer.yaml](./config_dereverb-echo_mel_band_roformer.yaml)<br>
+Model_256_8_4: [dereverb-echo_mel_band_roformer_sdr_10.0169.ckpt](./dereverb-echo_mel_band_roformer_sdr_10.0169.ckpt)<br>
+Instr dry sdr: 13.1507, Instr other sdr: 6.8830, Metric avg sdr: 10.0169
+
+Configs_128_4_4: [config_dereverb-echo_128_4_4_mel_band_roformer.yaml](./config_dereverb-echo_128_4_4_mel_band_roformer.yaml)<br>
+Model_128_4: [dereverb-echo_128_4_4_mel_band_roformer_sdr_dry_12.4235.ckpt](./dereverb-echo_128_4_4_mel_band_roformer_sdr_dry_12.4235.ckpt)
+Instr dry sdr: 12.4235
+
 Instruments: [dry, other]<br>
 Finetuned from: `model_mel_band_roformer_ep_3005_sdr_11.4360.ckpt`<br>
 Datasets: 
@@ -22,20 +28,6 @@ Datasets:
 - Validation datasets: 30 songs from my own collection
 - All random reverbs and delay effects are generated by [this python script](./scripts/create_reverb_delay.py) and sorted into the mustb18 dataset format.
 
-Metrics: Based on the sdr value of 30 songs for validation.
-
-```
-Instr dry sdr: 13.1507 (Std: 4.1088)
-Instr dry l1_freq: 53.7715 (Std: 13.3363)
-Instr dry si_sdr: 12.7707 (Std: 4.6134)
-Instr other sdr: 6.8830 (Std: 2.5547)
-Instr other l1_freq: 52.7358 (Std: 11.8587)
-Instr other si_sdr: 5.9448 (Std: 2.8721)
-Metric avg sdr        : 10.0169
-Metric avg l1_freq    : 53.2536
-Metric avg si_sdr     : 9.3577
-```
-
 ### Training log
 
 Training logs: [train.log](./train.log)<br>