mirror of
https://github.com/babysor/MockingBird.git
synced 2024-11-21 01:13:57 +08:00
cd20d21f3d
* add docker support * 修复训练集解压问题 * 修复web.py启动问题 * 混合数据集训练参数
18 lines
533 B
Docker
18 lines
533 B
Docker
FROM pytorch/pytorch:latest
|
|
|
|
RUN apt-get update && apt-get install -y build-essential ffmpeg parallel aria2 && apt-get clean
|
|
|
|
COPY ./requirements.txt /workspace/requirements.txt
|
|
|
|
RUN pip install -r requirements.txt && pip install webrtcvad-wheels
|
|
|
|
COPY . /workspace
|
|
|
|
VOLUME [ "/datasets", "/workspace/synthesizer/saved_models/" ]
|
|
|
|
ENV DATASET_MIRROR=default FORCE_RETRAIN=false TRAIN_DATASETS=aidatatang_200zh\ magicdata\ aishell3\ data_aishell TRAIN_SKIP_EXISTING=true
|
|
|
|
EXPOSE 8080
|
|
|
|
ENTRYPOINT [ "/workspace/docker-entrypoint.sh" ]
|