parent
2b44f374c1
commit
abb15ac6e8
@ -1,39 +1,49 @@
|
|||||||
data:
|
# https://yaml.org/type/float.html
|
||||||
data_dir: '/PATH/TO/DATA/hey_snips_research_6k_en_train_eval_clean_ter'
|
###########################################
|
||||||
dataset: 'paddleaudio.datasets:HeySnips'
|
# Data #
|
||||||
|
###########################################
|
||||||
|
dataset: 'paddleaudio.datasets:HeySnips'
|
||||||
|
data_dir: '/PATH/TO/DATA/hey_snips_research_6k_en_train_eval_clean_ter'
|
||||||
|
|
||||||
model:
|
############################################
|
||||||
num_keywords: 1
|
# Network Architecture #
|
||||||
backbone: 'paddlespeech.kws.models:MDTC'
|
############################################
|
||||||
config:
|
backbone: 'paddlespeech.kws.models:MDTC'
|
||||||
stack_num: 3
|
num_keywords: 1
|
||||||
stack_size: 4
|
stack_num: 3
|
||||||
in_channels: 80
|
stack_size: 4
|
||||||
res_channels: 32
|
in_channels: 80
|
||||||
kernel_size: 5
|
res_channels: 32
|
||||||
|
kernel_size: 5
|
||||||
|
|
||||||
feature:
|
###########################################
|
||||||
feat_type: 'kaldi_fbank'
|
# Feature #
|
||||||
sample_rate: 16000
|
###########################################
|
||||||
frame_shift: 10
|
feat_type: 'kaldi_fbank'
|
||||||
frame_length: 25
|
sample_rate: 16000
|
||||||
n_mels: 80
|
frame_shift: 10
|
||||||
|
frame_length: 25
|
||||||
|
n_mels: 80
|
||||||
|
|
||||||
training:
|
###########################################
|
||||||
epochs: 100
|
# Training #
|
||||||
num_workers: 16
|
###########################################
|
||||||
batch_size: 100
|
epochs: 100
|
||||||
checkpoint_dir: './checkpoint'
|
num_workers: 16
|
||||||
save_freq: 10
|
batch_size: 100
|
||||||
log_freq: 10
|
checkpoint_dir: './checkpoint'
|
||||||
learning_rate: 0.001
|
save_freq: 10
|
||||||
weight_decay: 0.00005
|
log_freq: 10
|
||||||
grad_clip: 5.0
|
learning_rate: 0.001
|
||||||
|
weight_decay: 0.00005
|
||||||
|
grad_clip: 5.0
|
||||||
|
|
||||||
scoring:
|
###########################################
|
||||||
batch_size: 100
|
# Scoring #
|
||||||
num_workers: 16
|
###########################################
|
||||||
checkpoint: './checkpoint/epoch_100/model.pdparams'
|
batch_size: 100
|
||||||
score_file: './scores.txt'
|
num_workers: 16
|
||||||
stats_file: './stats.0.txt'
|
checkpoint: './checkpoint/epoch_100/model.pdparams'
|
||||||
img_file: './det.png'
|
score_file: './scores.txt'
|
||||||
|
stats_file: './stats.0.txt'
|
||||||
|
img_file: './det.png'
|
@ -1,2 +1,25 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
python3 ${BIN_DIR}/plot_det_curve.py --cfg_path=$1 --keyword HeySnips
|
# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
# you may not use this file except in compliance with the License.
|
||||||
|
# You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
# See the License for the specific language governing permissions and
|
||||||
|
# limitations under the License.
|
||||||
|
|
||||||
|
if [ $# != 3 ];then
|
||||||
|
echo "usage: ${0} config_path checkpoint output_file"
|
||||||
|
exit -1
|
||||||
|
fi
|
||||||
|
|
||||||
|
keyword=$1
|
||||||
|
stats_file=$2
|
||||||
|
img_file=$3
|
||||||
|
|
||||||
|
python3 ${BIN_DIR}/plot_det_curve.py --keyword_label ${keyword} --stats_file ${stats_file} --img_file ${img_file}
|
||||||
|
@ -1,5 +1,27 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
# you may not use this file except in compliance with the License.
|
||||||
|
# You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
# See the License for the specific language governing permissions and
|
||||||
|
# limitations under the License.
|
||||||
|
|
||||||
python3 ${BIN_DIR}/score.py --cfg_path=$1
|
if [ $# != 4 ];then
|
||||||
|
echo "usage: ${0} checkpoint score_file stats_file"
|
||||||
|
exit -1
|
||||||
|
fi
|
||||||
|
|
||||||
python3 ${BIN_DIR}/compute_det.py --cfg_path=$1
|
cfg_path=$1
|
||||||
|
ckpt=$2
|
||||||
|
score_file=$3
|
||||||
|
stats_file=$4
|
||||||
|
|
||||||
|
python3 ${BIN_DIR}/score.py --config ${cfg_path} --ckpt ${ckpt} --score_file ${score_file} || exit -1
|
||||||
|
python3 ${BIN_DIR}/compute_det.py --config ${cfg_path} --score_file ${score_file} --stats_file ${stats_file} || exit -1
|
||||||
|
@ -1,13 +1,31 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
# you may not use this file except in compliance with the License.
|
||||||
|
# You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
# See the License for the specific language governing permissions and
|
||||||
|
# limitations under the License.
|
||||||
|
|
||||||
|
if [ $# != 2 ];then
|
||||||
|
echo "usage: ${0} num_gpus config_path"
|
||||||
|
exit -1
|
||||||
|
fi
|
||||||
|
|
||||||
ngpu=$1
|
ngpu=$1
|
||||||
cfg_path=$2
|
cfg_path=$2
|
||||||
|
|
||||||
if [ ${ngpu} -gt 0 ]; then
|
if [ ${ngpu} -gt 0 ]; then
|
||||||
python3 -m paddle.distributed.launch --gpus $CUDA_VISIBLE_DEVICES ${BIN_DIR}/train.py \
|
python3 -m paddle.distributed.launch --gpus $CUDA_VISIBLE_DEVICES ${BIN_DIR}/train.py \
|
||||||
--cfg_path ${cfg_path}
|
--config ${cfg_path}
|
||||||
else
|
else
|
||||||
echo "set CUDA_VISIBLE_DEVICES to enable multi-gpus trainning."
|
echo "set CUDA_VISIBLE_DEVICES to enable multi-gpus trainning."
|
||||||
python3 ${BIN_DIR}/train.py \
|
python3 ${BIN_DIR}/train.py \
|
||||||
--cfg_path ${cfg_path}
|
--config ${cfg_path}
|
||||||
fi
|
fi
|
||||||
|
Loading…
Reference in new issue