course_ner / docker-compose.yml
PravinKonze321's picture
Upload folder using huggingface_hub
a05b328 verified
raw
history blame
664 Bytes
version: "3.7"
services:
model-server:
image: your-docker-image-name
environment:
BASIC_AUTH_USER: "konze"
BASIC_AUTH_PASS: "konze@123"
LOG_LEVEL: "info"
WORKERS: 4
THREADS: 2
BASELINE_MODEL_NAME: "PravinKonze321/course_ner"
FINETUNED_MODEL_NAME: "finetuned_model"
LABEL_STUDIO_HOST: "http://192.168.2.88:8080"
LABEL_STUDIO_API_KEY: "d42beb6660a41e1baa13ad0b10c3547efbc5f264"
START_TRAINING_EACH_N_UPDATES: 10
LEARNING_RATE: 0.001
NUM_TRAIN_EPOCHS: 10
WEIGHT_DECAY: 0.01
MODEL_DIR: "./results"
volumes:
- ./results:/app/results
ports:
- "9090:9090"