mirror of
https://github.com/babysor/MockingBird.git
synced 2024-11-21 01:13:57 +08:00
cd20d21f3d
* add docker support * 修复训练集解压问题 * 修复web.py启动问题 * 混合数据集训练参数
24 lines
566 B
YAML
24 lines
566 B
YAML
version: '3.8'
|
|
|
|
services:
|
|
server:
|
|
image: mockingbird:latest
|
|
build: .
|
|
volumes:
|
|
- ./datasets:/datasets
|
|
- ./synthesizer/saved_models:/workspace/synthesizer/saved_models
|
|
environment:
|
|
- DATASET_MIRROR=US
|
|
- FORCE_RETRAIN=false
|
|
- TRAIN_DATASETS=aidatatang_200zh magicdata aishell3 data_aishell
|
|
- TRAIN_SKIP_EXISTING=true
|
|
ports:
|
|
- 8080:8080
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
device_ids: [ '0' ]
|
|
capabilities: [ gpu ]
|