fixed a bug in GPU deployment script

imedslab · Nov 27, 2019 · 88e0694 · 88e0694
1 parent 8c4d2e8
commit 88e0694
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/docker/docker-compose-gpu.yml b/docker/docker-compose-gpu.yml
@@ -32,9 +32,12 @@ services:
         source: ../snapshots_knee_grading/ # The snapshots are stored in the root directory
         target: /snapshots/
         read_only: true
+      - type: bind
+        source: ../logs
+        target: /logs/
     environment:
       - KNEEL_ADDR=http://kneel:5000
     entrypoint: ["python", "-m", "ouludeepknee.inference.app",
                    "--snapshots_path", "/snapshots/",
                    "--device", "cuda", "--deploy", "True",
-                 "--port", "5001", "--logs", "/logs/deepknee-cpu.log"]
+                 "--port", "5001", "--logs", "/logs/deepknee-gpu.log"]