synthesize_e2e.sh 676 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
#!/bin/bash

config_path=$1
train_output_path=$2
ckpt_name=$3
add_blank=$4

stage=0
stop_stage=0


if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then
    FLAGS_allocator_strategy=naive_best_fit \
    FLAGS_fraction_of_gpu_memory_to_use=0.01 \
    python3 ${BIN_DIR}/synthesize_e2e.py \
小湉湉's avatar
小湉湉 已提交
16
        --am=vits_aishell3 \
17 18 19 20 21 22
        --config=${config_path} \
        --ckpt=${train_output_path}/checkpoints/${ckpt_name} \
        --phones_dict=dump/phone_id_map.txt \
        --speaker_dict=dump/speaker_id_map.txt \
        --spk_id=0 \
        --output_dir=${train_output_path}/test_e2e \
H
Hui Zhang 已提交
23
        --text=${BIN_DIR}/../../assets/sentences.txt \
24 25
        --add-blank=${add_blank}
fi