Spaces:
Sleeping
Sleeping
File size: 1,448 Bytes
cbf01bb d450991 cbf01bb d450991 d643e60 d450991 cbf01bb d450991 cbf01bb | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 | version: "3.8"
services:
bert_classifier:
container_name: bert_classifier
image: heartexlabs/label-studio-ml-backend:bertclass-master
build:
context: .
args:
TEST_ENV: ${TEST_ENV}
environment:
# Essential Training Settings
# Controls how much the model updates its weights in response to errors
# - Lower (1e-5): More stable but slower learning
# - Higher (3e-5): Faster learning but might be unstable
# - Default (2e-5): Good balance for BERT fine-tuning
- LEARNING_RATE=2e-5
# Number of complete passes through the training data
# - Lower (1): Faster training but might underfit
# - Higher (3+): Better learning but might overfit
# - Default (2): Good balance for small datasets
- NUM_TRAIN_EPOCHS=3
# Prevents model weights from growing too large (regularization)
# - Lower (0.001): Less regularization, might overfit
# - Higher (0.1): More regularization, might underfit
# - Default (0.01): Standard value for BERT fine-tuning
- WEIGHT_DECAY=0.01
# Number of annotations before starting training
# - Set to 1 for testing (train after each annotation)
# - Recommended (5-10) for production (more stable training)
- START_TRAINING_EACH_N_UPDATES=1
ports:
- "9090:9090"
volumes:
- "./data/server:/data"
- "./data/.cache:/root/.cache"
|