ClearVoice / config /inference /SpEx_plus_TSE_8K.yaml
alibabasglab's picture
Upload 161 files
8e8cd3e verified
raw
history blame
746 Bytes
#!/bin/bash
mode: 'inference'
use_cuda: 1 # 1 for True, 0 for False
num_gpu: 1
sampling_rate: 8000
network: "SpEx_plus_TSE_8K" ##network type
checkpoint_dir: "checkpoints/SpEx_plus_TSE_8K"
input_path: "data/wsj0_2mix_16k_fullpath.lst" # an input dir or input scp file
#input_path: "/home/shengkui.zhao/DingTalk_NS/data/webrtc_test_set_20200521_16k/noisy"
#input_path: "/mnt/nas_sg/mit_sg/shengkui.zhao/ComplexNN/audio/youtube_testset_16k/noisy_long/noisy"
output_dir: "outputs/MossFormer2_SS_16K_wsj0_2mix" ## output dir to store processed audio
# decode parameters
one_time_decode_length: 5 #maximum segment length for one-pass decoding (seconds), longer audio (>3s) will use segmented decoding
decode_window: 1 #one-pass decoding length