forked from jyhan03/channel-decorrelation
-
Notifications
You must be signed in to change notification settings - Fork 0
/
decode.sh
44 lines (35 loc) · 771 Bytes
/
decode.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#!/bin/bash
set -eu
stage=1 # control the decoding stage
checkpoint=./exp/conv_tasnet/cd_adap/
gpuid=-1 # whether to use the gpu
data_root=./data/tt
mix_scp=$data_root/mix.scp
ref_scp=$data_root/ref.scp
aux_scp=$data_root/aux.scp
cal_sdr=1
fs=8000
dump_dir=$checkpoint/dump_dir
if [ $stage -le 1 ]; then
./nnet/evaluate.py \
--checkpoint $checkpoint \
--gpuid $gpuid \
--mix_scp $mix_scp \
--ref_scp $ref_scp \
--aux_scp $aux_scp \
--cal_sdr $cal_sdr \
> $1.eva.log 2>&1
fi
exit 1
if [ $stage -le 2 ]; then
./nnet/separate.py \
--checkpoint $checkpoint \
--gpuid $gpuid \
--mix_scp $mix_scp \
--ref_scp $ref_scp \
--aux_scp $aux_scp \
--fs $fs \
--dump-dir $dump_dir \
> $1.sep.log 2>&1
fi
echo "Separate done!"