DPDFNet4-CoreML / config.json
iky1e's picture
Publish validated DPDFNet-4 Core ML models
6866fdc verified
Raw
History Blame Contribute Delete
1.06 kB
{
"model_version": "DPDFNet4",
"dpdfnet_profile": "16khz",
"sample_rate": 16000,
"fft_size": 320,
"hop_size": 160,
"min_nb_erb_freqs": 1,
"nb_erb": 32,
"nb_df": 96,
"dprnn_num_blocks": 4,
"dprnn_erb_bins": 8,
"dprnn_df_bins": 48,
"encoder_erb_strides": [
1,
2,
2,
1
],
"decoder_erb_strides": [
1,
2,
2
],
"feature_type": "erb",
"mask_type": "erb",
"df_order": 5,
"df_lookahead": 2,
"conv_lookahead": 2,
"conv_ch": 64,
"emb_hidden_dim": 256,
"emb_num_layers": 2,
"df_hidden_dim": 256,
"df_num_layers": 2,
"gru_groups": 1,
"linear_groups": 16,
"enc_linear_groups": 32,
"group_shuffle": false,
"enc_concat": true,
"df_gru_skip": "groupedlinear",
"conv_kernel": [
1,
3
],
"convt_kernel": [
1,
3
],
"conv_kernel_inp": [
3,
3
],
"df_pathway_kernel_size_t": 5,
"lsnr_max": 35,
"lsnr_min": -15,
"decoder_upsample_type": "subpixel",
"source_checkpoint": "dpdfnet4.pth",
"source_repository": "https://github.com/ceva-ip/DPDFNet"
}