diff --git a/docker_trainer/Dockerfile b/docker_trainer/Dockerfile index 6d0589db..03f57817 100644 --- a/docker_trainer/Dockerfile +++ b/docker_trainer/Dockerfile @@ -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 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 tqdm==4.64.1 RUN pip install librosa==0.9.2 -RUN pip install torch==1.13.1 -RUN pip install torchaudio==0.13.1 +#RUN pip install librosa==0.8.1 RUN pip install psutil==5.9.4 +#RUN pip install psutil==5.4.8 RUN pip install tensorboard==2.11.2 RUN pip install pytz==2022.7.1 RUN pip install pyworld==0.3.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.1.0 RUN pip install matplotlib==3.6.3 +#RUN pip install matplotlib==3.2.2 ADD /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/ +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 -# 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 - - diff --git a/docker_trainer/README.md b/docker_trainer/README.md index 5a396fd1..69df18b0 100644 --- a/docker_trainer/README.md +++ b/docker_trainer/README.md @@ -51,6 +51,10 @@ batch sizeは適宜調整 $ cp configs_org/baseconfig.json configs/ $ python3 normalize.py True $ 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 ``` diff --git a/docker_trainer/start_trainer.sh b/docker_trainer/start_trainer.sh index f1fa6fe2..c1f794b4 100644 --- a/docker_trainer/start_trainer.sh +++ b/docker_trainer/start_trainer.sh @@ -1,7 +1,7 @@ #!/bin/bash set -eu -DOCKER_IMAGE=dannadori/trainer:20230201_111603 +DOCKER_IMAGE=dannadori/trainer:20230205_043458 docker run --gpus all --rm -ti \ -v `pwd`/trainer/dataset:/MMVC_Trainer/dataset \ @@ -15,5 +15,3 @@ docker run --gpus all --rm -ti \ $DOCKER_IMAGE /bin/bash - -