diff --git a/start.sh b/start.sh index 5d4f1649..a72e7b65 100644 --- a/start.sh +++ b/start.sh @@ -10,6 +10,4 @@ docker run -it --gpus all --shm-size=2g \ -v `pwd`/vc_resources:/resources \ -e LOCAL_UID=$(id -u $USER) \ -e LOCAL_GID=$(id -g $USER) \ - -p 6008:6006 -p 8081:8080 mmvc_trainer_docker "$@" - - + -p 6006:6006 -p 8080:8080 dannadori/voice-changer:20220823_085004 "$@" diff --git a/trainer/Dockerfile b/trainer/Dockerfile index c3c52335..39af8ac0 100644 --- a/trainer/Dockerfile +++ b/trainer/Dockerfile @@ -48,6 +48,7 @@ COPY fine_model/D_180000.pth /MMVC_Trainer/fine_model/D_180000.pth ### Copy from base COPY --from=base --chmod=777 /usr/local/lib/python3.9/dist-packages /usr/local/lib/python3.9/dist-packages COPY --from=base --chmod=777 /MMVC_Trainer /MMVC_Trainer +RUN chmod 0777 /MMVC_Trainer WORKDIR /MMVC_Trainer ADD /setup.sh /MMVC_Trainer/ diff --git a/trainer/exec.sh b/trainer/exec.sh index f2e21c64..2cd08d9a 100644 --- a/trainer/exec.sh +++ b/trainer/exec.sh @@ -40,8 +40,10 @@ voice_change_flag=false config= model= +escape_flag=false + # オプション解析 -while getopts tb:rvm:c:h OPT; do +while getopts tb:rvc:m:hx OPT; do case $OPT in t) training_flag=true @@ -64,13 +66,15 @@ while getopts tb:rvm:c:h OPT; do h | \?) usage && exit 1 ;; - + x) + escape_flag=true esac done -## コマンドライン引数から、オプション引数分を削除 -# shift $((OPTIND - 1)) + +# ## コマンドライン引数から、オプション引数分を削除 +# # shift $((OPTIND - 1)) # モード解析 if $training_flag && $voice_change_flag; then @@ -80,6 +84,8 @@ elif $training_flag; then echo "■■■ ト レ ー ニ ン グ モ ー ド ■■■" elif $voice_change_flag; then echo "■■■ ボ イ チ ェ ン モ ー ド ■■■" +elif $escape_flag; then + /bin/bash else warn "-t(トレーニングモード) と -v(ボイチェンモード)のいずれかを指定してください。" exit 1 @@ -87,41 +93,41 @@ fi -if $training_flag; then +# if $training_flag; then - python3 create_dataset_jtalk.py -f train_config -s 24000 -m dataset/multi_speaker_correspondence.txt - # date_tag=`date +%Y%m%d%H%M%S` - sed -ie 's/80000/8000/' train_ms.py - sed -ie "s/\"batch_size\": 10/\"batch_size\": $batch_size/" configs/train_config.json +# python3 create_dataset_jtalk.py -f train_config -s 24000 -m dataset/multi_speaker_correspondence.txt +# # date_tag=`date +%Y%m%d%H%M%S` +# sed -ie 's/80000/8000/' train_ms.py +# sed -ie "s/\"batch_size\": 10/\"batch_size\": $batch_size/" configs/train_config.json - python3 -m tensorboard.main --logdir logs --port 6006 --host 0.0.0.0 & +# python3 -m tensorboard.main --logdir logs --port 6006 --host 0.0.0.0 & - if ${resume_flag}; then - echo "トレーニング再開。バッチサイズ: ${batch_size}。" - python3 train_ms.py -c configs/train_config.json -m vc - else - echo "トレーニング開始。バッチサイズ: ${batch_size}。" - python3 train_ms.py -c configs/train_config.json -m vc -fg fine_model/G_180000.pth -fd fine_model/D_180000.pth - fi -fi +# if ${resume_flag}; then +# echo "トレーニング再開。バッチサイズ: ${batch_size}。" +# python3 train_ms.py -c configs/train_config.json -m vc +# else +# echo "トレーニング開始。バッチサイズ: ${batch_size}。" +# python3 train_ms.py -c configs/train_config.json -m vc -fg fine_model/G_180000.pth -fd fine_model/D_180000.pth +# fi +# fi -if $voice_change_flag; then - if [[ -z "$config" ]]; then - warn "コンフィグファイル(-c)を指定してください" - fi - if [[ -z "$model" ]]; then - warn "モデルファイル(-m)を指定してください" - fi +# if $voice_change_flag; then +# if [[ -z "$config" ]]; then +# warn "コンフィグファイル(-c)を指定してください" +# fi +# if [[ -z "$model" ]]; then +# warn "モデルファイル(-m)を指定してください" +# fi - cd /voice-changer-internal/voice-change-service +# cd /voice-changer-internal/voice-change-service - cp -r /resources/* . - if [[ -e ./setting.json ]]; then - cp ./setting.json ../frontend/dist/assets/setting.json - fi - echo "-----------!!" - echo $config $model - echo $model - python3 serverSIO.py 8080 $config $model -fi +# cp -r /resources/* . +# if [[ -e ./setting.json ]]; then +# cp ./setting.json ../frontend/dist/assets/setting.json +# fi +# echo "-----------!!" +# echo $config $model +# echo $model +# python3 serverSIO.py 8080 $config $model +# fi diff --git a/trainer/setup.sh b/trainer/setup.sh index 364ec539..44c48377 100644 --- a/trainer/setup.sh +++ b/trainer/setup.sh @@ -15,6 +15,6 @@ groupmod -g $GROUP_ID user #su user # echo "parameter: $@" -# exec /usr/sbin/gosu user /bin/bash exec.sh "$@" -/bin/bash +exec /usr/sbin/gosu user /bin/bash exec.sh "$@" +#/bin/bash