[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