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" | |