22 lines
509 B
YAML
22 lines
509 B
YAML
|
|
services:
|
|
evaluate:
|
|
build:
|
|
dockerfile: Dockerfile
|
|
context: evaluate
|
|
container_name: evaluate
|
|
# deploy:
|
|
# resources:
|
|
# reservations:
|
|
# devices:
|
|
# - driver: nvidia
|
|
# count: all
|
|
# capabilities: [gpu]
|
|
environment:
|
|
model_folder: "tuned"
|
|
dataset_folder: "/app/dataset_folder"
|
|
output_folder: "/app/output_folder"
|
|
volumes:
|
|
- ./out_folder:/app/dataset_folder
|
|
- ./models:/app/output_folder
|