mirror of
https://github.com/w-okada/voice-changer.git
synced 2025-01-23 05:25:01 +03:00
315 lines
9.7 KiB
Bash
315 lines
9.7 KiB
Bash
#!/bin/bash
|
||
set -eu
|
||
|
||
DOCKER_IMAGE=dannadori/voice-changer:20221028_220138
|
||
#DOCKER_IMAGE=voice-changer
|
||
|
||
|
||
MODE=$1
|
||
PARAMS=${@:2:($#-1)}
|
||
|
||
### DEFAULT VAR ###
|
||
DEFAULT_EX_PORT=18888
|
||
DEFAULT_USE_GPU=on # on|off
|
||
DEFAULT_VERBOSE=off # on|off
|
||
|
||
### ENV VAR ###
|
||
EX_PORT=${EX_PORT:-${DEFAULT_EX_PORT}}
|
||
USE_GPU=${USE_GPU:-${DEFAULT_USE_GPU}}
|
||
VERBOSE=${VERBOSE:-${DEFAULT_VERBOSE}}
|
||
|
||
#echo $EX_PORT $USE_GPU $VERBOSE
|
||
|
||
### INTERNAL SETTING ###
|
||
TENSORBOARD_PORT=6006
|
||
SIO_PORT=8080
|
||
|
||
|
||
###
|
||
if [ "${MODE}" = "MMVC_TRAIN" ]; then
|
||
echo "トレーニングを開始します"
|
||
|
||
docker run -it --gpus all --shm-size=128M \
|
||
-v `pwd`/exp/${name}/dataset:/MMVC_Trainer/dataset \
|
||
-v `pwd`/exp/${name}/logs:/MMVC_Trainer/logs \
|
||
-v `pwd`/exp/${name}/filelists:/MMVC_Trainer/filelists \
|
||
-v `pwd`/vc_resources:/resources \
|
||
-e LOCAL_UID=$(id -u $USER) \
|
||
-e LOCAL_GID=$(id -g $USER) \
|
||
-e EX_IP="`hostname -I`" \
|
||
-e EX_PORT=${EX_PORT} \
|
||
-e VERBOSE=${VERBOSE} \
|
||
-p ${EX_PORT}:6006 $DOCKER_IMAGE "$@"
|
||
|
||
elif [ "${MODE}" = "MMVC" ]; then
|
||
if [ "${USE_GPU}" = "on" ]; then
|
||
echo "MMVCを起動します(with gpu)"
|
||
|
||
docker run -it --gpus all --shm-size=128M \
|
||
-v `pwd`/vc_resources:/resources \
|
||
-e LOCAL_UID=$(id -u $USER) \
|
||
-e LOCAL_GID=$(id -g $USER) \
|
||
-e EX_IP="`hostname -I`" \
|
||
-e EX_PORT=${EX_PORT} \
|
||
-e VERBOSE=${VERBOSE} \
|
||
-p ${EX_PORT}:8080 $DOCKER_IMAGE "$@"
|
||
else
|
||
echo "MMVCを起動します(only cpu)"
|
||
docker run -it --shm-size=128M \
|
||
-v `pwd`/vc_resources:/resources \
|
||
-e LOCAL_UID=$(id -u $USER) \
|
||
-e LOCAL_GID=$(id -g $USER) \
|
||
-e EX_IP="`hostname -I`" \
|
||
-e EX_PORT=${EX_PORT} \
|
||
-e VERBOSE=${VERBOSE} \
|
||
-p ${EX_PORT}:8080 $DOCKER_IMAGE "$@"
|
||
|
||
# docker run -it --shm-size=128M \
|
||
# -v `pwd`/vc_resources:/resources \
|
||
# -e LOCAL_UID=$(id -u $USER) \
|
||
# -e LOCAL_GID=$(id -g $USER) \
|
||
# -e EX_IP="`hostname -I`" \
|
||
# -e EX_PORT=${EX_PORT} \
|
||
# -e VERBOSE=${VERBOSE} \
|
||
# --entrypoint="" \
|
||
# -p ${EX_PORT}:8080 $DOCKER_IMAGE /bin/bash
|
||
|
||
fi
|
||
|
||
elif [ "${MODE}" = "SOFT_VC" ]; then
|
||
if [ "${USE_GPU}" = "on" ]; then
|
||
echo "Start Soft-vc"
|
||
|
||
docker run -it --gpus all --shm-size=128M \
|
||
-v `pwd`/vc_resources:/resources \
|
||
-e LOCAL_UID=$(id -u $USER) \
|
||
-e LOCAL_GID=$(id -g $USER) \
|
||
-e EX_IP="`hostname -I`" \
|
||
-e EX_PORT=${EX_PORT} \
|
||
-e VERBOSE=${VERBOSE} \
|
||
-p ${EX_PORT}:8080 $DOCKER_IMAGE "$@"
|
||
else
|
||
echo "Start Soft-vc withou GPU is not supported"
|
||
fi
|
||
|
||
else
|
||
echo "
|
||
usage:
|
||
$0 <MODE> <params...>
|
||
MODE: select one of ['MMVC_TRAIN', 'MMVC', 'SOFT_VC']
|
||
" >&2
|
||
fi
|
||
|
||
|
||
|
||
|
||
# echo $EX_PORT
|
||
|
||
|
||
# echo "------"
|
||
# echo "$@"
|
||
# echo "------"
|
||
|
||
# # usage() {
|
||
# # echo "
|
||
# # usage:
|
||
# # For training
|
||
# # $0 [-t] -n <exp_name> [-b batch_size] [-r]
|
||
# # -t: トレーニングモードで実行する場合に指定してください。(train)
|
||
# # -n: トレーニングの名前です。(name)
|
||
# # -b: バッチサイズです。(batchsize)
|
||
# # -r: トレーニング再開の場合に指定してください。(resume)
|
||
# # For changing voice
|
||
# # $0 [-v] [-c config] [-m model] [-g on/off]
|
||
# # -v: ボイスチェンジャーモードで実行する場合に指定してください。(voice changer)
|
||
# # -c: トレーニングで使用したConfigのファイル名です。(config)
|
||
# # -m: トレーニング済みのモデルのファイル名です。(model)
|
||
# # -g: GPU使用/不使用。デフォルトはonなのでGPUを使う場合は指定不要。(gpu)
|
||
# # -p: port番号
|
||
# # For help
|
||
# # $0 [-h]
|
||
# # -h: show this help
|
||
# # " >&2
|
||
# # }
|
||
# # warn () {
|
||
# # echo "! ! ! $1 ! ! !"
|
||
# # exit 1
|
||
# # }
|
||
|
||
|
||
# # training_flag=false
|
||
# # name=999_exp
|
||
# # batch_size=10
|
||
# # resume_flag=false
|
||
|
||
# # voice_change_flag=false
|
||
# # config=
|
||
# # model=
|
||
# # gpu=on
|
||
# # port=8080
|
||
# # escape_flag=false
|
||
|
||
# # # オプション解析
|
||
# # while getopts tn:b:rvc:m:g:p:hx OPT; do
|
||
# # case $OPT in
|
||
# # t)
|
||
# # training_flag=true
|
||
# # ;;
|
||
# # n)
|
||
# # name="$OPTARG"
|
||
# # ;;
|
||
# # b)
|
||
# # batch_size="$OPTARG"
|
||
# # ;;
|
||
# # r)
|
||
# # resume_flag=true
|
||
# # ;;
|
||
# # v)
|
||
# # voice_change_flag=true
|
||
# # ;;
|
||
# # c)
|
||
# # config="$OPTARG"
|
||
# # ;;
|
||
# # m)
|
||
# # model="$OPTARG"
|
||
# # ;;
|
||
# # g)
|
||
# # gpu="$OPTARG"
|
||
# # ;;
|
||
# # p)
|
||
# # port="$OPTARG"
|
||
# # ;;
|
||
# # h | \?)
|
||
# # usage && exit 1
|
||
# # ;;
|
||
# # x)
|
||
# # escape_flag=true
|
||
# # esac
|
||
# # done
|
||
|
||
|
||
# # # モード解析
|
||
# # if $training_flag && $voice_change_flag; then
|
||
# # warn "-t（トレーニングモード） と -v（ボイチェンモード）は同時に指定できません。"
|
||
# # elif $training_flag; then
|
||
# # echo "■■■ ト レ ー ニ ン グ モ ー ド ■■■"
|
||
# # elif $voice_change_flag; then
|
||
# # echo "■■■ ボ イ チ ェ ン モ ー ド ■■■"
|
||
# # elif $escape_flag; then
|
||
# # /bin/bash
|
||
# # else
|
||
# # warn "-t（トレーニングモード） と -v（ボイチェンモード）のいずれかを指定してください。"
|
||
# # fi
|
||
|
||
# if [ "${MODE}" = "MMVC_TRAIN_INITIAL" ]; then
|
||
# echo "トレーニングを開始します"
|
||
# elif [ "${MODE}" = "MMVC" ]; then
|
||
# echo "MMVCを起動します"
|
||
|
||
# docker run -it --gpus all --shm-size=128M \
|
||
# -v `pwd`/vc_resources:/resources \
|
||
# -e LOCAL_UID=$(id -u $USER) \
|
||
# -e LOCAL_GID=$(id -g $USER) \
|
||
# -e EX_IP="`hostname -I`" \
|
||
# -e EX_PORT=${port} \
|
||
# -p ${port}:8080 $DOCKER_IMAGE -v -c ${config} -m ${model}
|
||
|
||
# elif [ "${MODE}" = "MMVC_VERBOSE" ]; then
|
||
# echo "MMVCを起動します(verbose)"
|
||
# elif [ "${MODE}" = "MMVC_CPU" ]; then
|
||
# echo "MMVCを起動します(CPU)"
|
||
# elif [ "${MODE}" = "MMVC_CPU_VERBOSE" ]; then
|
||
# echo "MMVCを起動します(CPU)(verbose)"
|
||
# elif [ "${MODE}" = "SOFT_VC" ]; then
|
||
# echo "Start Soft-vc"
|
||
# elif [ "${MODE}" = "SOFT_VC_VERBOSE" ]; then
|
||
# echo "Start Soft-vc(verbose)"
|
||
# else
|
||
# echo "
|
||
# usage:
|
||
# $0 <MODE> <params...>
|
||
# EX_PORT:
|
||
# MODE: one of ['MMVC_TRAIN', 'MMVC', 'SOFT_VC']
|
||
|
||
# For 'MMVC_TRAIN':
|
||
# $0 MMVC_TRAIN_INITIAL -n <exp_name> [-b batch_size] [-r]
|
||
# -n: トレーニングの名前です。(name)
|
||
# -b: バッチサイズです。(batchsize)
|
||
# -r: トレーニング再開の場合に指定してください。(resume)
|
||
# For 'MMVC'
|
||
# $0 MMVC [-c config] [-m model] [-g on/off] [-p port] [-v]
|
||
# -c: トレーニングで使用したConfigのファイル名です。(config)
|
||
# -m: トレーニング済みのモデルのファイル名です。(model)
|
||
# -g: GPU使用/不使用。デフォルトはonなのでGPUを使う場合は指定不要。(gpu)
|
||
# -p: Docker からExposeするport番号
|
||
# -v: verbose
|
||
# For 'SOFT_VC'
|
||
# $0 SOFT_VC [-c config] [-m model] [-g on/off]
|
||
# -p: port exposed from docker container.
|
||
# -v: verbose
|
||
# " >&2
|
||
# fi
|
||
|
||
|
||
|
||
# # if $training_flag; then
|
||
# # if $resume_flag; then
|
||
# # echo "トレーニングを再開します"
|
||
# # docker run -it --gpus all --shm-size=128M \
|
||
# # -v `pwd`/exp/${name}/dataset:/MMVC_Trainer/dataset \
|
||
# # -v `pwd`/exp/${name}/logs:/MMVC_Trainer/logs \
|
||
# # -v `pwd`/exp/${name}/filelists:/MMVC_Trainer/filelists \
|
||
# # -v `pwd`/vc_resources:/resources \
|
||
# # -e LOCAL_UID=$(id -u $USER) \
|
||
# # -e LOCAL_GID=$(id -g $USER) \
|
||
# # -p ${TENSORBOARD_PORT}:6006 $DOCKER_IMAGE -t -b ${batch_size} -r
|
||
# # else
|
||
# # echo "トレーニングを開始します"
|
||
# # docker run -it --gpus all --shm-size=128M \
|
||
# # -v `pwd`/exp/${name}/dataset:/MMVC_Trainer/dataset \
|
||
# # -v `pwd`/exp/${name}/logs:/MMVC_Trainer/logs \
|
||
# # -v `pwd`/exp/${name}/filelists:/MMVC_Trainer/filelists \
|
||
# # -v `pwd`/vc_resources:/resources \
|
||
# # -e LOCAL_UID=$(id -u $USER) \
|
||
# # -e LOCAL_GID=$(id -g $USER) \
|
||
# # -p ${TENSORBOARD_PORT}:6006 $DOCKER_IMAGE -t -b ${batch_size}
|
||
# # fi
|
||
# # fi
|
||
|
||
# # if $voice_change_flag; then
|
||
# # if [[ -z "$config" ]]; then
|
||
# # warn "コンフィグファイル(-c)を指定してください"
|
||
# # fi
|
||
# # if [[ -z "$model" ]]; then
|
||
# # warn "モデルファイル(-m)を指定してください"
|
||
# # fi
|
||
# # if [ "${gpu}" = "on" ]; then
|
||
# # echo "GPUをマウントして起動します。"
|
||
|
||
# # docker run -it --gpus all --shm-size=128M \
|
||
# # -v `pwd`/vc_resources:/resources \
|
||
# # -e LOCAL_UID=$(id -u $USER) \
|
||
# # -e LOCAL_GID=$(id -g $USER) \
|
||
# # -e EX_IP="`hostname -I`" \
|
||
# # -e EX_PORT=${port} \
|
||
# # -p ${port}:8080 $DOCKER_IMAGE -v -c ${config} -m ${model}
|
||
# # elif [ "${gpu}" = "off" ]; then
|
||
# # echo "CPUのみで稼働します。GPUは使用できません。"
|
||
# # docker run -it --shm-size=128M \
|
||
# # -v `pwd`/vc_resources:/resources \
|
||
# # -e LOCAL_UID=$(id -u $USER) \
|
||
# # -e LOCAL_GID=$(id -g $USER) \
|
||
# # -e EX_IP="`hostname -I`" \
|
||
# # -e EX_PORT=${port} \
|
||
# # -p ${port}:8080 $DOCKER_IMAGE -v -c ${config} -m ${model}
|
||
# # else
|
||
# # echo ${gpu}
|
||
# # warn "-g は onかoffで指定して下さい。"
|
||
|
||
# # fi
|
||
|
||
|
||
# # fi
|
||
|
||
|