HoneyTian's picture
update
25ec0f4
raw
history blame contribute delete
804 Bytes
model_name: "spectrum_unet_irm"
# spec
sample_rate: 8000
n_fft: 512
win_length: 200
hop_length: 80
spec_bins: 256
# model
conv_channels: 64
conv_kernel_size_input:
- 3
- 3
conv_kernel_size_inner:
- 1
- 3
conv_lookahead: 0
convt_kernel_size_inner:
- 1
- 3
embedding_hidden_size: 256
encoder_combine_op: "concat"
encoder_emb_skip_op: "none"
encoder_emb_linear_groups: 16
encoder_emb_hidden_size: 256
encoder_linear_groups: 32
lsnr_max: 30
lsnr_min: -15
norm_tau: 1.
decoder_emb_num_layers: 3
decoder_emb_skip_op: "none"
decoder_emb_linear_groups: 16
decoder_emb_hidden_size: 256
df_decoder_hidden_size: 256
df_num_layers: 2
df_order: 5
df_bins: 96
df_gru_skip: "grouped_linear"
df_decoder_linear_groups: 16
df_pathway_kernel_size_t: 5
df_lookahead: 2
# runtime
use_post_filter: true