noblebarkrr commited on
Commit
d921a3d
·
verified ·
1 Parent(s): 189420a

Upload folder using huggingface_hub

Browse files
mdx23c/mdx23c_sfx_jasper.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:527f0a98e106fc4e2ad939a8c09d4f59966aa13272c31a530b36598f81c0edd3
3
+ size 1344401799
mdx23c/mdx23c_sfx_jasper_config.yaml ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ audio:
2
+ chunk_size: 261120
3
+ dim_f: 4096
4
+ dim_t: 256
5
+ hop_length: 1024
6
+ n_fft: 8192
7
+ num_channels: 2
8
+ sample_rate: 44100
9
+ min_mean_abs: 0.001
10
+
11
+ model:
12
+ act: gelu
13
+ bottleneck_factor: 4
14
+ growth: 128
15
+ norm: InstanceNorm
16
+ num_blocks_per_scale: 2
17
+ num_channels: 128
18
+ num_scales: 5
19
+ num_subbands: 4
20
+ scale:
21
+ - 2
22
+ - 2
23
+
24
+ training:
25
+ batch_size: 6
26
+ gradient_accumulation_steps: 1
27
+ grad_clip: 0
28
+ instruments:
29
+ - foreground
30
+ - background
31
+ lr: 9.0e-05
32
+ patience: 2
33
+ reduce_factor: 0.95
34
+ target_instrument: null
35
+ num_epochs: 1000
36
+ num_steps: 1000
37
+ augmentation: false # enable augmentations by audiomentations and pedalboard
38
+ augmentation_type: simple1
39
+ use_mp3_compress: false # Deprecated
40
+ augmentation_mix: true # Mix several stems of the same type with some probability
41
+ augmentation_loudness: true # randomly change loudness of each stem
42
+ augmentation_loudness_type: 1 # Type 1 or 2
43
+ augmentation_loudness_min: 0.5
44
+ augmentation_loudness_max: 1.5
45
+ q: 0.95
46
+ coarse_loss_clip: true
47
+ ema_momentum: 0.999
48
+ optimizer: adam
49
+ other_fix: true # it's needed for checking on multisong dataset if other is actually instrumental
50
+
51
+ inference:
52
+ batch_size: 1
53
+ dim_t: 256
54
+ num_overlap: 4
mel_band_roformer/mbr_bgm_jasper.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:49248d3e106e95f351b4a51942b87c00b6c3d3f6c0da11f6cf4d86b2d8ba55f9
3
+ size 2739398173
mel_band_roformer/mbr_bgm_jasper_config.yaml ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ audio:
2
+ chunk_size: 352800
3
+ dim_f: 1024
4
+ dim_t: 256
5
+ hop_length: 441
6
+ n_fft: 2048
7
+ num_channels: 2
8
+ sample_rate: 44100
9
+ min_mean_abs: 0.000
10
+
11
+ model:
12
+ dim: 384
13
+ depth: 6
14
+ stereo: true
15
+ num_stems: 1
16
+ time_transformer_depth: 1
17
+ freq_transformer_depth: 1
18
+ num_bands: 60
19
+ dim_head: 64
20
+ heads: 8
21
+ attn_dropout: 0
22
+ ff_dropout: 0
23
+ flash_attn: True
24
+ dim_freqs_in: 1025
25
+ sample_rate: 44100 # needed for mel filter bank from librosa
26
+ stft_n_fft: 2048
27
+ stft_hop_length: 441
28
+ stft_win_length: 2048
29
+ stft_normalized: False
30
+ mask_estimator_depth: 2
31
+ multi_stft_resolution_loss_weight: 1.0
32
+ multi_stft_resolutions_window_sizes: !!python/tuple
33
+ - 4096
34
+ - 2048
35
+ - 1024
36
+ - 512
37
+ - 256
38
+ multi_stft_hop_size: 147
39
+ multi_stft_normalized: False
40
+
41
+ training:
42
+ batch_size: 2
43
+ gradient_accumulation_steps: 1
44
+ grad_clip: 0
45
+ instruments:
46
+ - vocals
47
+ - other
48
+ lr: 1.0e-05
49
+ patience: 2
50
+ reduce_factor: 0.95
51
+ target_instrument: vocals
52
+ num_epochs: 1000
53
+ num_steps: 1000
54
+ augmentation: false # enable augmentations by audiomentations and pedalboard
55
+ augmentation_type: null
56
+ use_mp3_compress: false # Deprecated
57
+ augmentation_mix: false # Mix several stems of the same type with some probability
58
+ augmentation_loudness: false # randomly change loudness of each stem
59
+ augmentation_loudness_type: 1 # Type 1 or 2
60
+ augmentation_loudness_min: 0
61
+ augmentation_loudness_max: 0
62
+ q: 0.95
63
+ coarse_loss_clip: false
64
+ ema_momentum: 0.999
65
+ optimizer: adam
66
+ other_fix: true # it's needed for checking on multisong dataset if other is actually instrumental
67
+ use_amp: true # enable or disable usage of mixed precision (float16) - usually it must be true
68
+
69
+ inference:
70
+ batch_size: 4
71
+ dim_t: 256
72
+ num_overlap: 2
mel_band_roformer/mbr_guitar_chencfd.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1a81a87b46eed215c051579ac5cff3b139bfdfb9fa3bf4666df059ef87404d14
3
+ size 276726667
mel_band_roformer/mbr_guitar_chencfd_config.yaml ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ audio:
2
+ chunk_size: 131584
3
+ dim_f: 1024
4
+ dim_t: 256
5
+ hop_length: 512
6
+ n_fft: 2048
7
+ num_channels: 2
8
+ sample_rate: 44100
9
+ min_mean_abs: 0.001
10
+
11
+ model:
12
+ dim: 192
13
+ depth: 8
14
+ stereo: true
15
+ num_stems: 1
16
+ time_transformer_depth: 1
17
+ freq_transformer_depth: 1
18
+ linear_transformer_depth: 0
19
+ num_bands: 60
20
+ dim_head: 64
21
+ heads: 8
22
+ attn_dropout: 0.1
23
+ ff_dropout: 0.1
24
+ flash_attn: True
25
+ dim_freqs_in: 1025
26
+ sample_rate: 44100 # needed for mel filter bank from librosa
27
+ stft_n_fft: 2048
28
+ stft_hop_length: 512
29
+ stft_win_length: 2048
30
+ stft_normalized: False
31
+ mask_estimator_depth: 2
32
+ multi_stft_resolution_loss_weight: 1.0
33
+ multi_stft_resolutions_window_sizes: !!python/tuple
34
+ - 4096
35
+ - 2048
36
+ - 1024
37
+ - 512
38
+ - 256
39
+ multi_stft_hop_size: 147
40
+ multi_stft_normalized: False
41
+ mlp_expansion_factor: 4 # Probably too big (requires a lot of memory for weights)
42
+ use_torch_checkpoint: False # it allows to greatly reduce GPU memory consumption during training (not fully tested)
43
+ skip_connection: False # Enable skip connection between transformer blocks - can solve problem with gradients and probably faster training
44
+
45
+ training:
46
+ batch_size: 7
47
+ gradient_accumulation_steps: 1
48
+ grad_clip: 0
49
+ instruments:
50
+ - guitar
51
+ - others
52
+ lr: 5.0e-05
53
+ patience: 2
54
+ reduce_factor: 0.95
55
+ target_instrument: guitar
56
+ num_epochs: 1000
57
+ num_steps: 1000
58
+ q: 0.95
59
+ coarse_loss_clip: true
60
+ ema_momentum: 0.999
61
+ optimizer: adam
62
+ other_fix: false # it's needed for checking on multisong dataset if other is actually instrumental
63
+ use_amp: true # enable or disable usage of mixed precision (float16) - usually it must be true
64
+
65
+ augmentations:
66
+ enable: true # enable or disable all augmentations (to fast disable if needed)
67
+ loudness: true # randomly change loudness of each stem on the range (loudness_min; loudness_max)
68
+ loudness_min: 0.5
69
+ loudness_max: 1.5
70
+
71
+ inference:
72
+ batch_size: 1
73
+ dim_t: 256
74
+ num_overlap: 4
scnet_masked/scnet_masked_choirsep_exp.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f4982b7e6facf41a81fbf00244cd20597fbbe2939b24731fcfe417651b62a5c
3
+ size 42522921
scnet_masked/scnet_masked_choirsep_exp_config.yaml ADDED
@@ -0,0 +1,152 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ audio:
2
+ chunk_size: 88200 # 44100 * 11
3
+ num_channels: 2
4
+ sample_rate: 44100
5
+ min_mean_abs: 0.000
6
+
7
+ model:
8
+ sources:
9
+ - soprano
10
+ - alto
11
+ - tenor
12
+ - bass
13
+ audio_channels: 2
14
+ dims:
15
+ - 4
16
+ - 32
17
+ - 64
18
+ - 128
19
+ nfft: 4096
20
+ hop_size: 1024
21
+ win_size: 4096
22
+ normalized: True
23
+ band_SR:
24
+ - 0.175
25
+ - 0.392
26
+ - 0.433
27
+ band_stride:
28
+ - 1
29
+ - 4
30
+ - 16
31
+ band_kernel:
32
+ - 3
33
+ - 4
34
+ - 16
35
+ conv_depths:
36
+ - 3
37
+ - 2
38
+ - 1
39
+ compress: 4
40
+ conv_kernel: 3
41
+ num_dplayer: 6
42
+ expand: 1
43
+
44
+ loss_multistft:
45
+ fft_sizes:
46
+ - 1024
47
+ - 2048
48
+ - 4096
49
+ hop_sizes:
50
+ - 512
51
+ - 1024
52
+ - 2048
53
+ win_lengths:
54
+ - 1024
55
+ - 2048
56
+ - 4096
57
+ window: "hann_window"
58
+ scale: "mel"
59
+ n_bins: 128
60
+ sample_rate: 44100
61
+ perceptual_weighting: true
62
+ w_sc: 1.0
63
+ w_log_mag: 1.0
64
+ w_lin_mag: 0.0
65
+ w_phs: 0.0
66
+ mag_distance: "L1"
67
+
68
+
69
+ training:
70
+ batch_size: 6
71
+ gradient_accumulation_steps: 1
72
+ grad_clip: 0
73
+ instruments:
74
+ - soprano
75
+ - alto
76
+ - tenor
77
+ - bass
78
+ lr: 4.0e-04
79
+ patience: 6
80
+ reduce_factor: 0.95
81
+ target_instrument: null
82
+ num_epochs: 1000
83
+ num_steps: 1000
84
+ q: 0.95
85
+ coarse_loss_clip: true
86
+ ema_momentum: 0.999
87
+ optimizer: adamw
88
+ normalize: false # perform normalization on input of model (use the same for inference!)
89
+ other_fix: false # it's needed for checking on multisong dataset if other is actually instrumental
90
+ use_amp: true # enable or disable usage of mixed precision (float16) - usually it must be true
91
+
92
+ augmentations:
93
+ enable: false # enable or disable all augmentations (to fast disable if needed)
94
+ loudness: true # randomly change loudness of each stem on the range (loudness_min; loudness_max)
95
+ loudness_min: 0.5
96
+ loudness_max: 1.5
97
+ mixup: false # mix several stems of same type with some probability (only works for dataset types: 1, 2, 3)
98
+ mixup_probs:
99
+ !!python/tuple # 2 additional stems of the same type (1st with prob 0.2, 2nd with prob 0.02)
100
+ - 0.2
101
+ - 0.02
102
+ mixup_loudness_min: 0.5
103
+ mixup_loudness_max: 1.5
104
+
105
+ # apply mp3 compression to mixture only (emulate downloading mp3 from internet)
106
+ mp3_compression_on_mixture: 0.01
107
+ mp3_compression_on_mixture_bitrate_min: 32
108
+ mp3_compression_on_mixture_bitrate_max: 320
109
+ mp3_compression_on_mixture_backend: "lameenc"
110
+
111
+ all:
112
+ channel_shuffle: 0.5 # Set 0 or lower to disable
113
+ random_inverse: 0.1 # inverse track (better lower probability)
114
+ random_polarity: 0.5 # polarity change (multiply waveform to -1)
115
+
116
+ mp3_compression: 0.01
117
+ mp3_compression_min_bitrate: 32
118
+ mp3_compression_max_bitrate: 320
119
+ mp3_compression_backend: "lameenc"
120
+
121
+ # pedalboard reverb block
122
+ pedalboard_reverb: 0.01
123
+ pedalboard_reverb_room_size_min: 0.1
124
+ pedalboard_reverb_room_size_max: 0.9
125
+ pedalboard_reverb_damping_min: 0.1
126
+ pedalboard_reverb_damping_max: 0.9
127
+ pedalboard_reverb_wet_level_min: 0.1
128
+ pedalboard_reverb_wet_level_max: 0.9
129
+ pedalboard_reverb_dry_level_min: 0.1
130
+ pedalboard_reverb_dry_level_max: 0.9
131
+ pedalboard_reverb_width_min: 0.9
132
+ pedalboard_reverb_width_max: 1.0
133
+
134
+ # pedalboard resample block
135
+ pedalboard_resample: 0.01
136
+ pedalboard_resample_target_sample_rate_min: 4000
137
+ pedalboard_resample_target_sample_rate_max: 44100
138
+
139
+ pitch_shift: 0.1
140
+ pitch_shift_min_semitones: -1
141
+ pitch_shift_max_semitones: 1
142
+
143
+ seven_band_parametric_eq: 0.1
144
+ seven_band_parametric_eq_min_gain_db: -9
145
+ seven_band_parametric_eq_max_gain_db: 9
146
+
147
+
148
+ inference:
149
+ batch_size: 8
150
+ dim_t: 256
151
+ num_overlap: 4
152
+ normalize: false