PublicMirrors/voice-changer
1
0
Fork 0
mirror of https://github.com/w-okada/voice-changer.git synced 2025-01-23 21:45:00 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

WIP: docker support v1.5.x

Browse Source
This commit is contained in:
wataru 2023-02-05 04:47:04 +09:00
parent 080fa52059
commit cd6edb14d3
3 changed files with 19 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
docker_trainer/Dockerfile
View File

@ -4,37 +4,37 @@ ARG DEBIAN_FRONTEND=noninteractive
RUN apt-get -y update && apt-get install -y emacs git curl python3-pip libsndfile1-dev ffmpeg RUN apt-get -y update && apt-get install -y emacs git curl python3-pip libsndfile1-dev ffmpeg
RUN git clone --depth 1 -b v1.5.0.0_SiFiGAN https://github.com/isletennos/MMVC_Trainer.git
ADD /model/D_v15_best.pth /MMVC_Trainer/fine_model/
ADD /model/G_v15_best.pth /MMVC_Trainer/fine_model/
RUN pip install torch==1.13.1
RUN pip install torchaudio==0.13.1
RUN pip install pydub==0.25.1 RUN pip install pydub==0.25.1
RUN pip install tqdm==4.64.1 RUN pip install tqdm==4.64.1
RUN pip install librosa==0.9.2 RUN pip install librosa==0.9.2
RUN pip install torch==1.13.1 #RUN pip install librosa==0.8.1
RUN pip install torchaudio==0.13.1
RUN pip install psutil==5.9.4 RUN pip install psutil==5.9.4
#RUN pip install psutil==5.4.8
RUN pip install tensorboard==2.11.2 RUN pip install tensorboard==2.11.2
RUN pip install pytz==2022.7.1 RUN pip install pytz==2022.7.1
RUN pip install pyworld==0.3.2 RUN pip install pyworld==0.3.2
RUN pip install retry==0.9.2 RUN pip install retry==0.9.2
RUN pip install numpy==1.21.6
RUN pip install h5py==3.8.0 RUN pip install h5py==3.8.0
#RUN pip install h5py==3.1.0
RUN pip install matplotlib==3.6.3 RUN pip install matplotlib==3.6.3
#RUN pip install matplotlib==3.2.2
ADD /warmup.py / ADD /warmup.py /
RUN python3 warmup.py RUN python3 warmup.py
RUN git clone --depth 1 -b v1.5.0.0_SiFiGAN https://github.com/isletennos/MMVC_Trainer.git
WORKDIR /MMVC_Trainer/ WORKDIR /MMVC_Trainer/
RUN git checkout 1109f1201e1ee9008ee61dc221d620eb1c93d7d0
ADD /model/D_v15_best.pth /MMVC_Trainer/fine_model/
ADD /model/G_v15_best.pth /MMVC_Trainer/fine_model/
RUN cp -r /MMVC_Trainer/configs /MMVC_Trainer/configs_org RUN cp -r /MMVC_Trainer/configs /MMVC_Trainer/configs_org
# python3 normalize.py True
# python3 create_dataset.py -f train_config -s 24000 -m dataset/multi_speaker_correspondence.txt
# tensorboard --logdir logs --port 5000
# python3 train_ms.py -c configs/train_config.json -m 20220306_24000 -fg fine_model/G_v15_best.pth -fd fine_model/D_v15_best.pth
#$ docker run --gpus all --rm -ti -v `pwd`/trainer/dataset:/MMVC_Trainer/dataset -v `pwd`/trainer/configs:/MMVC_Trainer/configs -v `pwd`/trainer/F0:/MMVC_Trainer/F0 -v `pwd`/trainer/cF0:/MMVC_Trainer/cF0 -v `pwd`/trainer/units:/MMVC_Trainer/units -v `pwd`/trainer/logs:/MMVC_Trainer/logs -v `pwd`/trainer/filelists:/MMVC_Trainer/filelists -p 5000:5000 trainer /bin/bash

6
docker_trainer/README.md
View File

@ -51,6 +51,10 @@ batch sizeは適宜調整
$ cp configs_org/baseconfig.json configs/ $ cp configs_org/baseconfig.json configs/
$ python3 normalize.py True $ python3 normalize.py True
$ python3 create_dataset.py -f train_config -s 24000 -m dataset/multi_speaker_correspondence.txt $ python3 create_dataset.py -f train_config -s 24000 -m dataset/multi_speaker_correspondence.txt
$ tensorboard --logdir logs --port 5000 $ tensorboard --logdir logs --port 5000 --bind_all &
# batch size 変更
$ python3 train_ms.py -c configs/train_config.json -m 20220306_24000 -fg fine_model/G_v15_best.pth -fd fine_model/D_v15_best.pth $ python3 train_ms.py -c configs/train_config.json -m 20220306_24000 -fg fine_model/G_v15_best.pth -fd fine_model/D_v15_best.pth
$ python3 train_ms.py -c configs/train_config.json -m 20220306_24000
``` ```

4
docker_trainer/start_trainer.sh
View File

@ -1,7 +1,7 @@
#!/bin/bash #!/bin/bash
set -eu set -eu
DOCKER_IMAGE=dannadori/trainer:20230201_111603 DOCKER_IMAGE=dannadori/trainer:20230205_043458
docker run --gpus all --rm -ti \ docker run --gpus all --rm -ti \
-v `pwd`/trainer/dataset:/MMVC_Trainer/dataset \ -v `pwd`/trainer/dataset:/MMVC_Trainer/dataset \
@ -15,5 +15,3 @@ docker run --gpus all --rm -ti \
$DOCKER_IMAGE /bin/bash $DOCKER_IMAGE /bin/bash