diff --git a/docker-compose.yml b/docker-compose.yml index 36a0b4a..1f63595 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -2,8 +2,8 @@ version: '3.8' services: perception-service: + image: ghcr.io/action-prediction-lab/pepper-perception:latest build: . - image: pepper-perception:latest shm_size: '2gb' # Required for PyTorch DataLoaders command: [ "python", "-m", "perception_service.main", "--backend", "combined" ] ports: