Implementation of a mock scheduler for jobs/tasks that will evolve into a job manager for running training jobs over vast.ai instances.
27 lines
434 B
YAML
27 lines
434 B
YAML
services:
|
|
scheduler:
|
|
build:
|
|
context: ..
|
|
dockerfile: docker/scheduler.Dockerfile
|
|
depends_on:
|
|
- worker-0
|
|
- worker-1
|
|
networks:
|
|
- fedavg
|
|
|
|
worker-0:
|
|
build:
|
|
context: ..
|
|
dockerfile: docker/worker-cpu.Dockerfile
|
|
networks:
|
|
- fedavg
|
|
|
|
worker-1:
|
|
build:
|
|
context: ..
|
|
dockerfile: docker/worker-cpu.Dockerfile
|
|
networks:
|
|
- fedavg
|
|
|
|
networks:
|
|
fedavg:
|