83 lines
2.5 KiB
YAML
83 lines
2.5 KiB
YAML
|
# docker-compose.tuning.yml - Ray Tune with all OCR services (PaddleOCR + DocTR)
|
||
|
|
# Usage:
|
||
|
|
# docker compose -f docker-compose.tuning.yml up -d paddle-ocr-gpu doctr-gpu
|
||
|
|
# docker compose -f docker-compose.tuning.yml run raytune --service paddle --samples 64
|
||
|
|
# docker compose -f docker-compose.tuning.yml run raytune --service doctr --samples 64
|
||
|
|
# docker compose -f docker-compose.tuning.yml down
|
||
|
|
#
|
||
|
|
# Note: EasyOCR uses port 8002 (same as PaddleOCR). Use docker-compose.tuning.easyocr.yml separately.
|
||
|
|
|
||
|
|
services:
|
||
|
|
raytune:
|
||
|
|
image: seryus.ddns.net/unir/raytune:latest
|
||
|
|
network_mode: host
|
||
|
|
shm_size: '5gb'
|
||
|
|
volumes:
|
||
|
|
- ./results:/app/results:rw
|
||
|
|
environment:
|
||
|
|
- PYTHONUNBUFFERED=1
|
||
|
|
|
||
|
|
paddle-ocr-gpu:
|
||
|
|
image: seryus.ddns.net/unir/paddle-ocr-gpu:latest
|
||
|
|
container_name: paddle-ocr-gpu-tuning
|
||
|
|
ports:
|
||
|
|
- "8002:8000"
|
||
|
|
volumes:
|
||
|
|
- ./dataset:/app/dataset:ro
|
||
|
|
- ./debugset:/app/debugset:rw
|
||
|
|
- paddlex-cache:/root/.paddlex
|
||
|
|
environment:
|
||
|
|
- PYTHONUNBUFFERED=1
|
||
|
|
- CUDA_VISIBLE_DEVICES=0
|
||
|
|
- PADDLE_DET_MODEL=PP-OCRv5_mobile_det
|
||
|
|
- PADDLE_REC_MODEL=PP-OCRv5_mobile_rec
|
||
|
|
deploy:
|
||
|
|
resources:
|
||
|
|
reservations:
|
||
|
|
devices:
|
||
|
|
- driver: nvidia
|
||
|
|
count: 1
|
||
|
|
capabilities: [gpu]
|
||
|
|
restart: unless-stopped
|
||
|
|
healthcheck:
|
||
|
|
test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://localhost:8000/health')"]
|
||
|
|
interval: 30s
|
||
|
|
timeout: 10s
|
||
|
|
retries: 3
|
||
|
|
start_period: 60s
|
||
|
|
|
||
|
|
doctr-gpu:
|
||
|
|
image: seryus.ddns.net/unir/doctr-gpu:latest
|
||
|
|
container_name: doctr-gpu-tuning
|
||
|
|
ports:
|
||
|
|
- "8003:8000"
|
||
|
|
volumes:
|
||
|
|
- ./dataset:/app/dataset:ro
|
||
|
|
- ./debugset:/app/debugset:rw
|
||
|
|
- doctr-cache:/root/.cache/doctr
|
||
|
|
environment:
|
||
|
|
- PYTHONUNBUFFERED=1
|
||
|
|
- CUDA_VISIBLE_DEVICES=0
|
||
|
|
- DOCTR_DET_ARCH=db_resnet50
|
||
|
|
- DOCTR_RECO_ARCH=crnn_vgg16_bn
|
||
|
|
deploy:
|
||
|
|
resources:
|
||
|
|
reservations:
|
||
|
|
devices:
|
||
|
|
- driver: nvidia
|
||
|
|
count: 1
|
||
|
|
capabilities: [gpu]
|
||
|
|
restart: unless-stopped
|
||
|
|
healthcheck:
|
||
|
|
test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://localhost:8000/health')"]
|
||
|
|
interval: 30s
|
||
|
|
timeout: 10s
|
||
|
|
retries: 3
|
||
|
|
start_period: 180s
|
||
|
|
|
||
|
|
volumes:
|
||
|
|
paddlex-cache:
|
||
|
|
name: paddlex-model-cache
|
||
|
|
doctr-cache:
|
||
|
|
name: doctr-model-cache
|