diff --git a/docker-compose.yml b/docker-compose.yml index 7ac9751..05b6ca7 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -94,23 +94,23 @@ services: capabilities: [ "gpu" ] driver: "nvidia" -# photo_review_6: -# <<: *template -# container_name: photo_review_6 -# hostname: photo_review_6 -# volumes: -# - ./log:/app/log -# - ./model:/app/model -# depends_on: -# - photo_review_5 -# command: [ "photo_review.py" ] -# deploy: -# resources: -# reservations: -# devices: -# - device_ids: [ "0", "1" ] -# capabilities: [ "gpu" ] -# driver: "nvidia" + photo_review_6: + <<: *template + container_name: photo_review_6 + hostname: photo_review_6 + volumes: + - ./log:/app/log + - ./model:/app/model + depends_on: + - photo_review_5 + command: [ "photo_review.py" ] + deploy: + resources: + reservations: + devices: + - device_ids: [ "0", "1" ] + capabilities: [ "gpu" ] + driver: "nvidia" photo_mask_1: <<: *template