Create config.ini

DeepFilterNet2/config.ini

@@ -0,0 +1,108 @@
+[train]
+seed = 43
+device = 
+model = deepfilternet2
+jit = false
+mask_only = false
+df_only = false
+batch_size = 96
+batch_size_eval = 128
+num_workers = 16
+max_sample_len_s = 3.0
+p_atten_lim = 0.0
+p_reverb = 0.1
+overfit = false
+max_epochs = 100
+log_freq = 100
+log_timings = False
+validation_criteria = loss
+validation_criteria_rule = min
+early_stopping_patience = 15
+global_ds_sampling_f = 1
+num_prefetch_batches = 8
+dataloader_snrs = -5,0,5,10,20,40
+detect_anomaly = false
+batch_size_scheduling = 0/8,1/16,2/24,5/32,10/64,20/128,40/9999
+start_eval = true
+validation_set_caching = false
+
+[df]
+sr = 48000
+fft_size = 960
+hop_size = 480
+nb_erb = 32
+nb_df = 96
+norm_tau = 1
+lsnr_max = 35
+lsnr_min = -15
+min_nb_erb_freqs = 2
+pad_mode = input_specf
+
+[deepfilternet]
+conv_lookahead = 2
+conv_ch = 64
+conv_depthwise = True
+emb_hidden_dim = 256
+emb_num_layers = 3
+gru_groups = 8
+linear_groups = 8
+conv_dec_mode = transposed
+convt_depthwise = True
+mask_pf = False
+df_order = 5
+df_lookahead = 2
+df_hidden_dim = 256
+df_num_layers = 2
+dfop_method = df
+group_shuffle = False
+conv_kernel = 1,3
+df_gru_skip = none
+df_output_layer = groupedlinear
+gru_type = squeeze
+df_pathway_kernel_size_t = 5
+df_n_iter = 1
+enc_concat = True
+conv_kernel_inp = 3,3
+
+[localsnrloss]
+factor = 1e-3
+
+[maskloss]
+factor = 0
+mask = iam
+gamma = 0.6
+gamma_pred = 0.6
+f_under = 1
+
+[spectralloss]
+factor_magnitude = 1000
+factor_complex = 1000
+gamma = 0.3
+
+[dfalphaloss]
+factor = 0.0
+
+[multiresspecloss]
+factor = 500
+factor_complex = 500
+gamma = 0.3
+fft_sizes = 256,512,1024
+
+[optim]
+lr = 0.001
+momentum = 0
+weight_decay = 1e-12
+weight_decay_end = 0.05
+optimizer = adamw
+lr_min = 1e-06
+lr_warmup = 0.0001
+warmup_epochs = 3
+lr_cycle_mul = 1.0
+lr_cycle_decay = 0.5
+lr_cycle_limit = 1
+lr_update_per_epoch = False
+lr_cycle_epochs = -1
+
+[sdrloss]
+factor = 0.0
+segmental_ws = 0