-
Notifications
You must be signed in to change notification settings - Fork 0
/
docker-compose.yml
30 lines (28 loc) · 963 Bytes
/
docker-compose.yml
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
# command docker-compose build --build-arg DIRPATH={DIRPATH} --build-arg HF_USER={HF_USER} --build-arg DATASET_VERSION={DATASET_VERSION}
# will trigger the use of DIRPATH for the DockerFile
version: "3.9"
services:
training_clf:
image: andret94/training_language_model-training_clf
volumes:
- ./:/workspace/data
environment:
- HF_USER=${HF_USER}
- DIRPATH=${DIRPATH}
- HF_TOKEN=${HF_TOKEN}
- GITHUB_USER=
- MODEL_VERSION_MLM=${MODEL_VERSION_MLM}
- MODEL_VERSION_CLF=${MODEL_VERSION_CLF}
- DATASET_VERSION=${DATASET_VERSION}
training_mlm:
image: andret94/training_language_model-training_mlm
volumes:
- ./:/workspace/data
environment:
- HF_USER=${HF_USER}
- DIRPATH=${DIRPATH}
- HF_TOKEN=${HF_TOKEN}
- GITHUB_USER=
- MODEL_VERSION_MLM=${MODEL_VERSION_MLM}
- MODEL_VERSION_CLF=${MODEL_VERSION_CLF}
- DATASET_VERSION=${DATASET_VERSION}