Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,012 Bytes
12bfd03 |
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 |
#!/bin/bash
source path.sh
set -e
log_root="logs"
# .lst save the wav path.
input_training_file="train.lst"
input_validation_file="valid.lst"
#mode=debug
mode=train
if [ "${mode}" == "debug" ]; then
## debug
echo "Debug"
log_root=${log_root}_debug
export CUDA_VISIBLE_DEVICES=0
python ${BIN_DIR}/train.py \
--config config_24k_320d.json \
--checkpoint_path ${log_root} \
--input_training_file ${input_training_file} \
--input_validation_file ${input_validation_file} \
--checkpoint_interval 100 \
--summary_interval 10 \
--validation_interval 100 \
elif [ "$mode" == "train" ]; then
## train
echo "Train model..."
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
python ${BIN_DIR}/train.py \
--config config_24k_320d.json \
--checkpoint_path ${log_root} \
--input_training_file ${input_training_file} \
--input_validation_file ${input_validation_file} \
--checkpoint_interval 5000 \
--summary_interval 100 \
--validation_interval 5000
fi
|